Error when loading from a saved checkpoint file

[PlumedSerpent]

Hi, I have a training script with structure as : model=Net() flops, params = get_model_complexity_info(model, (3, 256, 256), as_strings=True, print_per_layer_stat=True) print('Flops: ' + flops) print('Params: ' + params) training torch.save(model) Then when I tried to load from above saved checkpoint using torch.load('checkpoint.pth'), it occurs into an error: AttributeError: 'Net' object has no attribute 'start_flops_count'. Do you have any idea why I get this error and how can i solve it?

[sovrasov]

Hi! Please take a look to the pytorch guidelines for saving models: https://pytorch.org/docs/stable/notes/serialization.html#recommend-saving-models