torch.save(model) in dump_checkpoint
🚀 Feature
Current PL save the only weight of network, but param searched network change their architecture flexibly while training.
Trainer.(..., save_entier_model=True) could be option
Motivation

After training 'param searched network' with pl and closed session, recognized I cannot load weight! :(
Saving the only state_dict is reasonable in most cases, but give the option to users for saving the whole network will be nice in some cases.
Alternatives
Manually save by users
Additional context
Always thx for cool dev project!