valeriechen / ask-your-humans Goto Github PK

>>> import torch
>>> torch.load('trained_models/our_model.pt')
/anaconda/envs/mazebase3.6/lib/python3.6/site-packages/torch/serialization.py:434: SourceChangeWarning: source code of class 'pytorchppo.a2c_ppo_acktr.model.Policy' has changed. you can retrieve the original source code by accessing the object's source attribute or set `torch.nn.Module.dump_patches = True` and use the patch tool to revert the changes.
  warnings.warn(msg, SourceChangeWarning)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/anaconda/envs/mazebase3.6/lib/python3.6/site-packages/torch/serialization.py", line 367, in load
    return _load(f, map_location, pickle_module)
  File "/anaconda/envs/mazebase3.6/lib/python3.6/site-packages/torch/serialization.py", line 538, in _load
    result = unpickler.load()
AttributeError: Can't get attribute 'MazeBaseBase' on <module 'pytorchppo.a2c_ppo_acktr.model' from '/path/to/ask-your-humans/mazebase-training/pytorchppo/a2c_ppo_acktr/model.py'>