pickle. Load error [attributeerror: can't get attribute 'volatile' on ＜ module '\u main']

pickle.load Report errors 【AttributeError: Can’t get attribute ‘Vocabulary’ on <module ‘main’】

• background ： When looking at the code, I want to see a pkl The file structure of the file and the data inside , So I use pickle.load open , But there was AttributeError: Can't get attribute 'Vocabulary' on <module '__main__' error

• reason ： For example, there are two ways to save models , One is to save the whole model , Although very large, it is easy to use ; The other is to save only the parameters of the model , Here, you need to initialize a model before loading . It's the same here ,msvd_vocab.pkl When loading, you also need to find the class it depends on , And it is also explained in the error information , The classes to be imported are Vocabulary. So you need to find the class in the project Vocabulary, And then call pickle.load('.../msvd_vocab.pkl','rb') Of python In file import This kind

• Error code

import pickle

with open('.\data\MSVD\msvd_vocab.pkl','rb') as f:
    v = pickle.load(f)

#  Report errors ： stay demo.py Of main There is no ‘Vocabulary’ modular 
AttributeError: Can't get attribute 'Vocabulary' on <module '__main__' from '.../demo.py'>

• find ‘Vocabulary’ class
  
• correct

import pickle
from utils.utils import *

with open('.\data\MSVD\msvd_vocab.pkl','rb') as f:
    v = pickle.load(f)

Perfect flowers