You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Hi,
I've tried your sample. But I found that the code has some issues for tensor conversion
When it runs to these two lines, th will report 'invalid arguments CudaLongTensor CudaTensor'
lloss = lloss + YYb:eq(Yb):sum()
lloss = lloss + YYb:eq(Yb):sum(2):eq(5):sum()
I changed it into the following to make it work
lloss = lloss + YYb:long():eq(Yb:long()):sum()
lloss = lloss + YYb:long():eq(Yb:long()):sum(2):eq(5):sum()
Btw, the training for 10K samples are very slow in my GTX 1080. Do I need to set something else?
Thanks
Hi,
I've tried your sample. But I found that the code has some issues for tensor conversion
When it runs to these two lines, th will report 'invalid arguments CudaLongTensor CudaTensor'
lloss = lloss + YYb:eq(Yb):sum()
lloss = lloss + YYb:eq(Yb):sum(2):eq(5):sum()
I changed it into the following to make it work
lloss = lloss + YYb:long():eq(Yb:long()):sum()
lloss = lloss + YYb:long():eq(Yb:long()):sum(2):eq(5):sum()
Btw, the training for 10K samples are very slow in my GTX 1080. Do I need to set something else?
Thanks
train .. 0.011111111111111
loss..250.91162276268
valid .. 3.38
train .. 6.5777777777778
valid .. 0
train .. 0.044444444444444
loss..248.19494605064
valid .. 3.98
train .. 8.1777777777778
valid .. 0
train .. 0.21111111111111
The text was updated successfully, but these errors were encountered: