clara/pytorch-ai
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions

Trained

Browse source
This commit is contained in:
Clemens-Dautermann 2020-01-27 20:58:35 +01:00
parent 3ce16b7010
commit 55cff9b18f
287 changed files with 115778 additions and 177 deletions
Download patch file Download diff file Expand all files Collapse all files

16
wandb/run-20200127_141329-lscctqxi/output.log Normal file
View file

@ -0,0 +1,16 @@
runnning on cuda:0
0%| | 0/4000 [00:00<?, ?it/s]
Traceback (most recent call last):
File "mnist_classifier.py", line 59, in <module>
output = net(X.view(-1, 784)) # pass in the reshaped batch (recall they are 28x28 atm)
File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 532, in __call__
result = self.forward(*input, **kwargs)
File "mnist_classifier.py", line 34, in forward
x = F.relu(self.fc1(x))
File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 532, in __call__
result = self.forward(*input, **kwargs)
File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/linear.py", line 87, in forward
return F.linear(input, self.weight, self.bias)
File "/usr/local/lib/python3.6/dist-packages/torch/nn/functional.py", line 1370, in linear
ret = torch.addmm(bias, input, weight.t())
RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 'mat1' in call to _th_addmm