i can't save model on colab

[komalanadkat]

I tried to save this model on colab. model_save_name = 'classifier.pt' path = F"/content/gdrive/My Drive/{model_save_name}" torch.save(LMF.state_dict(), path)

but found error TypeError Traceback (most recent call last) in () 1 model_save_name = 'classifier.pt' 2 path = F"/content/gdrive/My Drive/{model_save_name}" ----> 3 torch.save(LMF.state_dict(), path)

TypeError: state_dict() missing 1 required positional argument: 'self'

[Justin1904]

Hi, from the code you showed it seems you are calling .state_dict method on the LMF class itself, rather than on a concrete instance of LMF class. You can refer to this PyTorch documentation on how to properly save checkpoints.