Save/load model for inference

you can set save_weights_only=True in ModelCheckpoint which will save the hparams and model.state_dict().
https://pytorch-lightning.readthedocs.io/en/latest/generated/pytorch_lightning.callbacks.ModelCheckpoint.html#pytorch_lightning.callbacks.ModelCheckpoint