Training on checkpointed model fails [BROKEN]
This commit is contained in:
parent
d0d69f610e
commit
03c5160495
@ -128,3 +128,6 @@ if __name__ == "__main__":
|
||||
new_model = GLVQIris.load_from_checkpoint(checkpoint_path=ckpt)
|
||||
|
||||
print(new_model)
|
||||
|
||||
# Continue training
|
||||
trainer.fit(new_model, train_loader) # TODO See why this fails!
|
||||
|
Loading…
Reference in New Issue
Block a user