Traceback (most recent call last):


 "demo.py", line 56, in <module>

demo(opt)


 "demo.py", line 49, in demo

ret = detector.run(image_name)


 "/workspace/midhilesh/CenterNet/src/lib/detectors/base_detector.py", line 116, in run

output, dets, forward_time = self.process(images, return_time=True)


 "/workspace/midhilesh/CenterNet/src/lib/detectors/ctdet.py", line 30, in process

output = self.model(images)[-1]


 "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__

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


 "/workspace/midhilesh/CenterNet/src/lib/models/networks/pose_dla_dcn.py", line 471, in forward

x = self.base(x)


 "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__

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


 "/workspace/midhilesh/CenterNet/src/lib/models/networks/pose_dla_dcn.py", line 288, in forward

x = self.base_layer(x)


 "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__

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


 "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/container.py", line 91, in forward

input = module(input)


 "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__

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


 "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 301, in forward

self.padding, self.dilation, self.groups)
RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS

Traceback (most recent call last):


 "./fastreid/engine/train_loop.py", line 126, in train

self.run_step()


 "./fastreid/engine/train_loop.py", line 213, in run_step

outs = self.model(data)


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

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


 "./fastreid/modeling/meta_arch/mgn.py", line 105, in forward

b1_outputs = self.b1_head(b1_feat, targets)


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

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


 "./fastreid/modeling/heads/embedding_head.py", line 83, in forward

cls_outputs = self.classifier(bn_feat, targets)


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

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


 "./fastreid/layers/circle_softmax.py", line 38, in forward

pred_class_logits = targets * s_p + (1.0 - targets) * s_n
RuntimeError: expected backend CUDA and dtype Float but got backend CUDA and dtype Lon
