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

Debug warmup state

Browse Source
This commit is contained in:
James Betker 2022-06-10 15:23:31 -06:00
parent 6d85fe05f6
commit ee2827dee9
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
codes/trainer/injectors/gaussian_diffusion_injector.py
View File

@ -56,6 +56,13 @@ class GaussianDiffusionInjector(Injector):
self.channel_balancing_fn = None self.channel_balancing_fn = None
assert k <= 1, 'Only one channel filtering function can be applied.' assert k <= 1, 'Only one channel filtering function can be applied.'
def extra_metrics(self):
if hasattr(self, 'schedule_sampler') and isinstance(self.schedule_sampler, LossSecondMomentResampler):
return {
'sampler_warmed_up': self.schedule_sampler._warmed_up()
}
return {}
def forward(self, state): def forward(self, state):
gen = self.env['generators'][self.opt['generator']] gen = self.env['generators'][self.opt['generator']]
hq = state[self.input] hq = state[self.input]