a-One-Fan/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

Fix saver in distributed mode

Browse Source
This commit is contained in:
James Betker 2021-06-14 09:41:06 -06:00
parent 545f2db170
commit 60079a1572
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/trainer/eval/sr_diffusion_fid.py
View File

@ -50,8 +50,8 @@ class SrDiffusionFidEvaluator(evaluator.Evaluator):
torch.distributed.all_gather(gather_list, gen) torch.distributed.all_gather(gather_list, gen)
gen = torch.cat(gather_list, dim=0) gen = torch.cat(gather_list, dim=0)
for b in range(self.batch_sz): for g in gen:
torchvision.utils.save_image(gen[b], osp.join(fid_fake_path, "%i_.png" % (counter))) torchvision.utils.save_image(g, osp.join(fid_fake_path, f"{counter}.png"))
counter += 1 counter += 1
return {"fid": fid_score.calculate_fid_given_paths([self.fid_real_samples, fid_fake_path], self.fid_batch_size, return {"fid": fid_score.calculate_fid_given_paths([self.fid_real_samples, fid_fake_path], self.fid_batch_size,