mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix gdi bug?

Browse Source
This commit is contained in:
James Betker 2021-12-07 09:53:48 -07:00
parent 6ccff3f49f
commit ef15a39841
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/injectors/gaussian_diffusion_injector.py
View File

@ -31,7 +31,7 @@ class GaussianDiffusionInjector(Injector):
hq = state[self.input]
# In eval mode, seed torch with a deterministic seed for reproducibility.
if not gen.trainable:
if not gen.training():
torch.manual_seed(0)
random.seed(0)
@ -51,7 +51,7 @@ class GaussianDiffusionInjector(Injector):
self.output_x_start_key: diffusion_outputs['x_start_predicted']})
# Absolutely critical to undo the above seed.
if not gen.trainable:
if not gen.training():
torch.manual_seed(int(time.time()))
random.seed(int(time.time()))