From ec4da0846ddaf77e8406ad2fd592a6f0404ce5ae Mon Sep 17 00:00:00 2001 From: Quentin Bertrand Date: Tue, 9 Apr 2024 09:46:52 -0400 Subject: [PATCH] DEFINE_float >> DEFINE_integer (#112) --- examples/images/cifar10/compute_fid.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/images/cifar10/compute_fid.py b/examples/images/cifar10/compute_fid.py index 2f79554..ffa66c2 100644 --- 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)