lambders / drl-experiments Goto Github PK

Could you tell me the version of your pytorch. I got runtime error:
python main.py --algo=ppo --mode=train
pygame 1.9.6
Hello from the pygame community. https://www.pygame.org/contribute.html
/workdata/git/drl-experiments/ppo.py:54: UserWarning: nn.init.uniform is now deprecated in favor of nn.init.uniform_.
torch.nn.init.uniform(m.weight, -0.01, 0.01)
libpng warning: iCCP: known incorrect sRGB profile
libpng warning: iCCP: known incorrect sRGB profile
libpng warning: iCCP: known incorrect sRGB profile
libpng warning: iCCP: known incorrect sRGB profile
libpng warning: iCCP: known incorrect sRGB profile
libpng warning: iCCP: known incorrect sRGB profile
libpng warning: iCCP: cHRM chunk does not match sRGB
libpng warning: iCCP: known incorrect sRGB profile
libpng warning: iCCP: known incorrect sRGB profile
libpng warning: iCCP: known incorrect sRGB profile
libpng warning: iCCP: known incorrect sRGB profile
libpng warning: iCCP: known incorrect sRGB profile
libpng warning: iCCP: known incorrect sRGB profile
libpng warning: iCCP: known incorrect sRGB profile
libpng warning: iCCP: known incorrect sRGB profile
libpng warning: iCCP: known incorrect sRGB profile
Traceback (most recent call last):
File "main.py", line 133, in
agent.train()
File "/workdata/git/drl-experiments/ppo.py", line 241, in train
values, actions, action_log_probs = self.net.act(states)
File "/workdata/git/drl-experiments/ppo.py", line 87, in act
values, action_logits = self.forward(x)
File "/workdata/git/drl-experiments/ppo.py", line 72, in forward
x = self.relu1(self.conv1(x))
File "/home/minshi/.virtualenvs/drl-experiments-0ZyJdSJ6/lib/python3.6/site-packages/torch/nn/modules/module.py", line 532, in call
result = self.forward(*input, **kwargs)
File "/home/minshi/.virtualenvs/drl-experiments-0ZyJdSJ6/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 345, in forward
return self.conv2d_forward(input, self.weight)
File "/home/minshi/.virtualenvs/drl-experiments-0ZyJdSJ6/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 342, in conv2d_forward
self.padding, self.dilation, self.groups)
RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same