Training on checkpointed model fails [BROKEN]

This commit is contained in:
Jensun Ravichandran 2021-04-22 11:56:54 +02:00
parent d0d69f610e
commit 03c5160495

View File

@ -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!