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
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:
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)