kxngluca/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

dont pickle unique files

Browse Source
This commit is contained in:
James Betker 2022-01-21 00:02:06 -07:00
parent 7fef7fb9ff
commit 91b4b240ac
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
codes/data/audio/fast_paired_dataset.py
View File

@ -182,8 +182,7 @@ class FastPairedVoiceDebugger:
def get_state(self): def get_state(self):
return {'total_items': self.total_items, return {'total_items': self.total_items,
'loaded_items': self.loaded_items, 'loaded_items': self.loaded_items,
'self_conditioning_items': self.self_conditioning_items, 'self_conditioning_items': self.self_conditioning_items}
'unique_files_loaded': self.unique_files}
def load_state(self, state): def load_state(self, state):
if isinstance(state, dict): if isinstance(state, dict):