[Questions:] What is the pytorch version for atari?

[gyber]

Hi, thanks for this great project!

I was wondering that which version of pytorch should I use for atari?
I use 0.2 and it works well for minirts.
But when I eval atari breakout.bin with pretrained model, these messages would appear:

Load from /home/ubuntu/atari_breakout.bin
/home/ubuntu/miniconda3/envs/elf/lib/python3.6/site-packages/torch/serialization.py:286: SourceChangeWarning: source code of class 'model.Model_ActorCritic' has changed. you can retieve 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 "eval.py", line 24, in
model = env["model_loaders"][0].load_model(GC.params)
File "/home/ubuntu/ELF/rlpytorch/model_loader.py", line 82, in load_model
model.load(self.load, omit_keys=omit_keys)
File "/home/ubuntu/ELF/rlpytorch/model_base.py", line 93, in load
data = torch.load(filename)
File "/home/ubuntu/miniconda3/envs/elf/lib/python3.6/site-packages/torch/serialization.py", line 231, in load
return _load(f, map_location, pickle_module)
File "/home/ubuntu/miniconda3/envs/elf/lib/python3.6/site-packages/torch/serialization.py", line 379, in _load
result = unpickler.load()
ModuleNotFoundError: No module named 'rlmethod_common'

I guess this is because I use a wrong pytorch version for atari?
And if I use a model trained by myself, it will just get stuck in 0%:
(I don't use a GPU for atari)

Running ROM file...
Random seed is 23468651
Action set: 0 1 3 4 11 12
Version: 1f790173095cd910976d9f651b80beb872ec5d12_staged
Num Actions: 6
#recv_thread = 4
Group 0:
Collector[0] Batchsize: 32 Info: [gid=0][T=1][name=""]
Collector[1] Batchsize: 32 Info: [gid=1][T=1][name=""]
Collector[2] Batchsize: 32 Info: [gid=2][T=1][name=""]
Collector[3] Batchsize: 32 Info: [gid=3][T=1][name=""]
Load from /home/ubuntu/save-831.bin
Loaded = /home/ubuntu/save-831.bin
0%| | 0/500 [13:24<?, ?it/s]