update PATH to compute fid

atong01 · Jan 16, 2024 · f7db0ef · f7db0ef
1 parent 952bdec
commit f7db0ef
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
@@ -47,7 +47,7 @@
 
 
 # Load the model
-PATH = f"{FLAGS.input_dir}/{FLAGS.model}/cifar10_weights_step_{FLAGS.step}.pt"
+PATH = f"{FLAGS.input_dir}/{FLAGS.model}/{FLAGS.model}_cifar10_weights_step_{FLAGS.step}.pt"
 print("path: ", PATH)
 checkpoint = torch.load(PATH)
 state_dict = checkpoint["ema_model"]