AttributeError: Can't get attribute 'CNN' on <module 'main'>

[minji-o-j]

AttributeError Traceback (most recent call last) in () 1 import torch ----> 2 model = torch.load ('/content/gdrive/My Drive/Colab Notebooks/my_model.pkl') 3 model
1 frames /usr/local/lib/python3.6/dist-packages/torch/serialization.py in _legacy_load(f, map_location, pickle_module, *
pickle_load_args) 771 unpickler = pickle_module.Unpickler(f, *pickle_load_args) 772 unpickler.persistent_load = persistent_load --> 773 result = unpickler.load() 774 775 deserialized_storage_keys = pickle_module.load(f, **pickle_load_args)
AttributeError: Can't get attribute 'CNN' on <module 'main'>

[zlh1992]

Is there any solution of this problem?
I just met the same problem
torch 1.5.1