DEFINE_float >> DEFINE_integer (#112)

atong01 · Apr 9, 2024 · ec4da08 · ec4da08
1 parent af8fec6
commit ec4da08
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/images/cifar10/compute_fid.py b/examples/images/cifar10/compute_fid.py
@@ -27,7 +27,7 @@
 flags.DEFINE_integer("step", 400000, help="training steps")
 flags.DEFINE_integer("num_gen", 50000, help="number of samples to generate")
 flags.DEFINE_float("tol", 1e-5, help="Integrator tolerance (absolute and relative)")
-flags.DEFINE_float("batch_size_fid", 1024, help="Batch size to compute FID")
+flags.DEFINE_integer("batch_size_fid", 1024, help="Batch size to compute FID")
 
 FLAGS(sys.argv)