add in predict

Author: davidsketchdeck

main_tpu.py

@@ -75,7 +75,8 @@ def parse_args():
 	parser.add_argument('--ld'             , type=float    , default=10.0          , help='The gradient penalty lambda')
 	parser.add_argument('--n-critic'       , type=int      , default=2             , help='The number of critic')
 
-	parser.add_argument('--inception-score-num'     , type=int      , default=50000            , help='The number of sample images to use in inception score')
+	# IGoodfellow says sould be 50k
+	parser.add_argument('--inception-score-num'     , type=int      , default=512            , help='The number of sample images to use in inception score')
 	parser.add_argument('--sample-num'     , type=int      , default=36            , help='The number of sample images to save')
 	parser.add_argument('--test-num'       , type=int      , default=10            , help='The number of images generated by the test')
 
@@ -249,8 +250,8 @@ def main():
 			logger.info(f"Generate predictions {epoch}")
 			predictions = tpu_estimator.predict(input_fn=predict_input_fn)
 
-			# logger.info(f"Save predictions")
-			# save_predictions(args, args.result_dir, eval_file, predictions, epoch, total_steps, experiment)
+			logger.info(f"Save predictions")
+			save_predictions(args, args.result_dir, eval_file, predictions, epoch, total_steps, experiment)