forked from mrq/DL-Art-School
Fix dvae bug
This commit is contained in:
parent
0799d95af5
commit
0237e96b34
|
@ -125,8 +125,11 @@ class DiscreteVAE(nn.Module):
|
||||||
return images
|
return images
|
||||||
|
|
||||||
def get_debug_values(self, step, __):
|
def get_debug_values(self, step, __):
|
||||||
# Report annealing schedule
|
if self.record_codes:
|
||||||
return {'histogram_codes': self.codes}
|
# Report annealing schedule
|
||||||
|
return {'histogram_codes': self.codes}
|
||||||
|
else:
|
||||||
|
return {}
|
||||||
|
|
||||||
@torch.no_grad()
|
@torch.no_grad()
|
||||||
@eval_decorator
|
@eval_decorator
|
||||||
|
|
Loading…
Reference in New Issue
Block a user