L.LightningModuleの子クラス想定

joblib.Memory

詳しく調べていないけど、直接だとキャッシュ使えなかったから、cloudpickle.dumpsで渡すようにした

チェックポイント

>If you don’t just want to load weights, but instead restore the full training, do the following:

https://lightning.ai/docs/pytorch/stable/common/checkpointing_basic.html#resume-training-state

普通にロードすればweightだけになるっぽい