From 7976a5825d5dd0a206e40b91b2b662214986bd5f Mon Sep 17 00:00:00 2001 From: James Betker Date: Fri, 1 Jan 2021 13:00:59 -0700 Subject: [PATCH] srfid is incorrectly labeled --- codes/trainer/eval/sr_fid.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/trainer/eval/sr_fid.py b/codes/trainer/eval/sr_fid.py index b5e6e097..ad70f47e 100644 --- a/codes/trainer/eval/sr_fid.py +++ b/codes/trainer/eval/sr_fid.py @@ -48,7 +48,7 @@ class SrFidEvaluator(evaluator.Evaluator): torchvision.utils.save_image(hq_gen_hf_applied[b], osp.join(fid_fake_path, "%i_.png" % (counter))) counter += 1 - return {"fid": fid_score.calculate_fid_given_paths([self.fid_real_samples, fid_fake_path], self.batch_sz, True, + return {"sr_fid": fid_score.calculate_fid_given_paths([self.fid_real_samples, fid_fake_path], self.batch_sz, True, 2048)}