alpha0422 / label-smoothing-cuda Goto Github PK

-> fert_loss=loss_func(fert_out, fert_label, args['fert_smooth'],\
(Pdb) loss_func(fert_out, fert_label, args['fert_smooth'],-1,False)
*** RuntimeError: input.dim() == 2 INTERNAL ASSERT FAILED at csrc/label_smoothing_cuda_kernel.cu:431, please report a bug to PyTorch. Currently only 2 dim input supported (host_softmax_xentropy at csrc/label_smoothing_cuda_kernel.cu:431)
frame #0: c10::Error::Error(c10::SourceLocation, std::string const&) + 0x47 (0x7ffa59955627 in /home/wzh/anaconda3/envs/lbz37/lib/python3.7/site-packages/torch/lib/libc10.so)
frame #1: std::vector<at::Tensor, std::allocator<at::Tensor> > host_softmax_xentropy<LogSoftMaxForwardEpilogue>(at::Tensor const&, at::Tensor const&, float, bool) + 0x503 (0x7ffa452459c2 in /home/wzh/anaconda3/envs/lbz37/lib/python3.7/site-packages/label_smoothing-0.1-py3.7-linux-x86_64.egg/label_smoothing_cuda.cpython-37m-x86_64-linux-gnu.so)
frame #2: softmax_xentropy_cuda(at::Tensor const&, at::Tensor const&, float, bool) + 0x50 (0x7ffa4523bff5 in /home/wzh/anaconda3/envs/lbz37/lib/python3.7/site-packages/label_smoothing-0.1-py3.7-linux-x86_64.egg/label_smoothing_cuda.cpython-37m-x86_64-linux-gnu.so)
frame #3: softmax_xentropy_forward(at::Tensor const&, at::Tensor const&, float, bool) + 0xa7 (0x7ffa4522d267 in /home/wzh/anaconda3/envs/lbz37/lib/python3.7/site-packages/label_smoothing-0.1-py3.7-linux-x86_64.egg/label_smoothing_cuda.cpython-37m-x86_64-linux-gnu.so)
frame #4: <unknown function> + 0x33d47 (0x7ffa45239d47 in /home/wzh/anaconda3/envs/lbz37/lib/python3.7/site-packages/label_smoothing-0.1-py3.7-linux-x86_64.egg/label_smoothing_cuda.cpython-37m-x86_64-linux-gnu.so)
frame #5: <unknown function> + 0x30f17 (0x7ffa45236f17 in /home/wzh/anaconda3/envs/lbz37/lib/python3.7/site-packages/label_smoothing-0.1-py3.7-linux-x86_64.egg/label_smoothing_cuda.cpython-37m-x86_64-linux-gnu.so)
<omitting python frames>
frame #11: THPFunction_apply(_object*, _object*) + 0xa0f (0x7ffa8fb35a3f in /home/wzh/anaconda3/envs/lbz37/lib/python3.7/site-packages/torch/lib/libtorch_python.so)
frame #60: __libc_start_main + 0xe7 (0x7ffaebda9b97 in /lib/x86_64-linux-gnu/libc.so.6)