From 9fbe5df9bde1013e193c648ed8247bf2cc0190d2 Mon Sep 17 00:00:00 2001 From: QB3 Date: Mon, 8 Apr 2024 09:53:27 -0400 Subject: [PATCH] DEFINE_float >> DEFINE_integer --- 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)