Question about load checkpoint for my own model

[exiawsh]

Hello, I have defined my Lightning module like this

But when I load the checkpoint, I got error, how can I load my own cfg and checkpoint corrcetly?

[rohitgr7]

hey @exiawsh !

two things here:

1. load_from_checkpoint is a class method so you need to call it using TrainingModule.load_from_checkpoint
2. Since cfg is not part of hyperparameters, you need to call it like this TrainingModule.load_from_checkpoint(checkpoint_path, cfg=cfg)

check out the note and examples here: https://pytorch-lightning.readthedocs.io/en/latest/common/lightning_module.html#load-from-checkpoint