deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

clip adf test dataset

Browse Source
This commit is contained in:
James Betker 2022-05-27 13:42:52 -06:00
parent 31dec016e0
commit 031769150d
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/trainer/eval/audio_diffusion_fid.py
View File

@ -33,6 +33,8 @@ class AudioDiffusionFid(evaluator.Evaluator):
super().__init__(model, opt_eval, env, uses_all_ddp=True) super().__init__(model, opt_eval, env, uses_all_ddp=True)
self.real_path = opt_eval['eval_tsv'] self.real_path = opt_eval['eval_tsv']
self.data = load_tsv_aligned_codes(self.real_path) self.data = load_tsv_aligned_codes(self.real_path)
if 'clip_dataset' in opt_eval.keys():
self.data = self.data[:opt_eval['clip_dataset']]
if distributed.is_initialized() and distributed.get_world_size() > 1: if distributed.is_initialized() and distributed.get_world_size() > 1:
self.skip = distributed.get_world_size() # One batch element per GPU. self.skip = distributed.get_world_size() # One batch element per GPU.
else: else: