I saved the object of my trainer class using the torch save method. However, when loading the saved file, the depickle process can’t find “__iterator”, which is a private method inside my “Trainer” class. Because of that, the whole loading process fails. Any idea how to solve or bypass it? I just need to recover the data saved into the attributes of that class.
Traceback (most recent call last): File ".conda/envs/py37/lib/python3.7/site-packages/IPython/core/interactiveshell.py", line 3331, in run_code exec(code_obj, self.user_global_ns, self.user_ns) File "<ipython-input-1-a5666d77c70f>", line 1, in <module> torch.load("snapshots/model.pth", map_location='cpu') File ".conda/envs/py37/lib/python3.7/site-packages/torch/serialization.py", line 529, in load return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args) File ".conda/envs/py37/lib/python3.7/site-packages/torch/serialization.py", line 702, in _legacy_load result = unpickler.load() AttributeError: 'Trainer' object has no attribute '__iterator'