Skip to content
Snippets Groups Projects
Commit 04d7a206 authored by Gonzalo Martin Garcia's avatar Gonzalo Martin Garcia
Browse files

solved minor bugs, added images to README

parent 056a9793
Branches
Tags v2.2.0
No related merge requests found
Show whitespace changes
Inline Side-by-side
Showing
with 1 addition and 0 deletions
trainer/train.py
+ 1
− 0
View file @ 04d7a206
...@@ -163,6 +163,7 @@ def ddpm_trainer(model, ...@@ -163,6 +163,7 @@ def ddpm_trainer(model,
ema.module.load_state_dict(checkpoint['ema']) ema.module.load_state_dict(checkpoint['ema'])
except Exception as e: except Exception as e:
print("Error loading checkpoint. Exception: ", e) print("Error loading checkpoint. Exception: ", e)
raise
# pick kl loss function # pick kl loss function
if model.kl_loss == 'weighted': if model.kl_loss == 'weighted':
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment