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

fix memory issue

Browse Source
This commit is contained in:
James Betker 2022-06-03 11:20:09 -06:00
parent 9d8c2bddb1
commit a14274c845
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
codes/train.py
View File

@ -180,6 +180,7 @@ class Trainer:
self.val_freq = int(opt['train']['val_freq_megasamples'] * 1000000) self.val_freq = int(opt['train']['val_freq_megasamples'] * 1000000)
self.next_eval_step = self.total_training_data_encountered + self.val_freq self.next_eval_step = self.total_training_data_encountered + self.val_freq
del resume_state # For whatever reason, this relieves a memory burden on the first GPU for some training sessions.
def do_step(self, train_data): def do_step(self, train_data):
if self._profile: if self._profile: