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)
|
new_model = GLVQIris.load_from_checkpoint(checkpoint_path=ckpt)
|
||||||
|
|
||||||
print(new_model)
|
print(new_model)
|
||||||
|
|
||||||
|
# Continue training
|
||||||
|
trainer.fit(new_model, train_loader) # TODO See why this fails!
|
||||||
|
Loading…
Reference in New Issue
Block a user