/home/amittel/Desktop/pytorch-semantic-segmentation-master/utils/misc.py:47: UserWarning: Implicit dimension choice for log_softmax has been deprecated. Change the call to include dim=X as an argument.
return self.nll_loss(F.log_softmax(inputs), targets)
Traceback (most recent call last):
File “train.py”, line 227, in
main(args)
File “train.py”, line 97, in main
train(train_loader, net, criterion, optimizer, epoch, train_args)
File “train.py”, line 122, in train
loss.backward()
File “/home/amittel/anaconda2/envs/pytorch_sem/lib/python3.6/site-packages/torch/autograd/variable.py”, line 167, in backward
torch.autograd.backward(self, gradient, retain_graph, create_graph, retain_variables)
File “/home/amittel/anaconda2/envs/pytorch_sem/lib/python3.6/site-packages/torch/autograd/init.py”, line 99, in backward
variables, grad_variables, retain_graph)
RuntimeError: CUDNN_STATUS_INTERNAL_ERROR
I have tried this solution, but it hasn’t worked so far.