Allow fit.py to start model from checkpoint

BRIO-lab / LitJTML

Using Pytorch Lightning and WandbLogger for our JTML neural network segementation code

1 stars 5 forks source link

Open sasank-desaraju opened 2 years ago

sasank-desaraju commented 2 years ago

Currently no way to train a model from a checkpoint. Allow to pull from config.module['CKPT_FILE'].