Traceback (most recent call last):


 "/snap/pycharm-community/132/helpers/pydev/pydevd.py", line 1758, in <module>

main()


 "/snap/pycharm-community/132/helpers/pydev/pydevd.py", line 1752, in main

globals = debugger.run(setup['file'], None, None, is_module)


 "/snap/pycharm-community/132/helpers/pydev/pydevd.py", line 1147, in run

pydev_imports.execfile(file, globals, locals)
# execute the script


 "/snap/pycharm-community/132/helpers/pydev/_pydev_imps/_pydev_execfile.py", line 18, in execfile

exec(compile(contents+"\n", file, 'exec'), glob, loc)


 "/home/user/maskrcnn-benchmark/tools/train_net.py", line 190, in <module>

main()


 "/home/user/maskrcnn-benchmark/tools/train_net.py", line 183, in main

model = train(cfg, args.local_rank, args.distributed)


 "/home/user/maskrcnn-benchmark/tools/train_net.py", line 88, in train

arguments,


 "/home/user/maskrcnn-benchmark/maskrcnn_benchmark/engine/trainer.py", line 85, in do_train

loss_dict = model(images, targets)


 "/home/user/anaconda3/envs/tf_python_37/lib/python3.7/site-packages/torch/nn/modules/module.py", line 493, in __call__

result = self.forward(*input, **kwargs)


 "/home/user/anaconda3/envs/tf_python_37/lib/python3.7/site-packages/apex-0.1-py3.7-linux-x86_64.egg/apex/amp/_initialize.py", line 193, in new_fwd

**applier(kwargs, input_caster))


 "/home/user/maskrcnn-benchmark/maskrcnn_benchmark/modeling/detector/generalized_rcnn.py", line 52, in forward

x, result, detector_losses = self.roi_heads(features, proposals, targets)


 "/home/user/anaconda3/envs/tf_python_37/lib/python3.7/site-packages/torch/nn/modules/module.py", line 493, in __call__

result = self.forward(*input, **kwargs)


 "/home/user/maskrcnn-benchmark/maskrcnn_benchmark/modeling/roi_heads/roi_heads.py", line 53, in forward

x, detections, loss_keypoint = self.keypoint(keypoint_features, detections, targets)


 "/home/user/anaconda3/envs/tf_python_37/lib/python3.7/site-packages/torch/nn/modules/module.py", line 493, in __call__

result = self.forward(*input, **kwargs)


 "/home/user/maskrcnn-benchmark/maskrcnn_benchmark/modeling/roi_heads/keypoint_head/keypoint_head.py", line 45, in forward

loss_kp = self.loss_evaluator(proposals, kp_logits)


 "/home/user/maskrcnn-benchmark/maskrcnn_benchmark/modeling/roi_heads/keypoint_head/loss.py", line 168, in __call__

keypoint_loss = F.cross_entropy(keypoint_logits[valid], keypoint_targets[valid])


 "/home/user/anaconda3/envs/tf_python_37/lib/python3.7/site-packages/torch/nn/functional.py", line 2056, in cross_entropy

return nll_loss(log_softmax(input, 1), target, weight, None, ignore_index, None, reduction)


 "/home/user/anaconda3/envs/tf_python_37/lib/python3.7/site-packages/torch/nn/functional.py", line 1871, in nll_loss

ret = torch._C._nn.nll_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index)
RuntimeError: cuda runtime error (59) : device-side assert triggered at /opt/conda/conda-

Traceback (most recent call last):


 "train.py", line 332, in <module>

model = train_model(model, criterion, optimizer_ft, exp_lr_scheduler,num_epochs=10)


 "train.py", line 176, in train_model

outputs = model(inputs)


 "/home/yhangbin/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 357, in __call__

result = self.forward(*input, **kwargs)


 "/home/yhangbin/code/github/Person_reID_baseline_pytorch/model.py", line 162, in forward

predict[i] = c(part[i])


 "/home/yhangbin/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 357, in __call__

result = self.forward(*input, **kwargs)


 "/home/yhangbin/code/github/Person_reID_baseline_pytorch/model.py", line 47, in forward

x = self.add_block(x)


 "/home/yhangbin/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 357, in __call__

result = self.forward(*input, **kwargs)


 "/home/yhangbin/anaconda3/lib/python3.6/site-packages/torch/nn/modules/container.py", line 67, in forward

input = module(input)


 "/home/yhangbin/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 357, in __call__

result = self.forward(*input, **kwargs)


 "/home/yhangbin/anaconda3/lib/python3.6/site-packages/torch/nn/modules/batchnorm.py", line 37, in forward

self.training, self.momentum, self.eps)


 "/home/yhangbin/anaconda3/lib/python3.6/site-packages/torch/nn/functional.py", line 1013, in batch_norm

return f(input, weight, bias)
RuntimeError: CHECK_ARG(tensor->nDimension >= 2 && tensor->nDimension <= 5) failed at torch/csrc/cudnn/BatchNorm.cpp:13
