Traceback (most recent call last):
File "./tools/train.py", line 190, in
main()
File "./tools/train.py", line 186, in main
meta=meta)
File "/home/syy/MixD/mmdet/apis/train.py", line 154, in train_detector
runner.run(data_loaders, cfg.workflow, cfg.total_epochs)
File "/home/syy/anaconda3/envs/fgd/lib/python3.7/site-packages/mmcv/runner/epoch_based_runner.py", line 125, in run
epoch_runner(data_loaders[i], **kwargs)
File "/home/syy/anaconda3/envs/fgd/lib/python3.7/site-packages/mmcv/runner/epoch_based_runner.py", line 50, in train
self.run_iter(data_batch, train_mode=True)
File "/home/syy/anaconda3/envs/fgd/lib/python3.7/site-packages/mmcv/runner/epoch_based_runner.py", line 30, in run_iter
**kwargs)
File "/home/syy/anaconda3/envs/fgd/lib/python3.7/site-packages/mmcv/parallel/data_parallel.py", line 67, in train_step
return self.module.train_step(*inputs[0], **kwargs[0])
File "/home/syy/MixD/mmdet/models/detectors/base.py", line 247, in train_step
losses = self(**data)
File "/home/syy/anaconda3/envs/fgd/lib/python3.7/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/syy/anaconda3/envs/fgd/lib/python3.7/site-packages/mmcv/runner/fp16_utils.py", line 84, in new_func
return old_func(*args, **kwargs)
File "/home/syy/MixD/mmdet/models/detectors/base.py", line 181, in forward
return self.forward_train(img, img_metas, **kwargs)
File "/home/syy/MixD/mmdet/distillation/distillers/detection_distiller.py", line 137, in forward_train
student_loss = self.student.forward_train(img, img_metas, **kwargs)
File "/home/syy/MixD/mmdet/models/detectors/kd_one_stage.py", line 73, in forward_train
teacher_x = self.teacher_model.extract_feat(img)
File "/home/syy/MixD/mmdet/models/detectors/single_stage.py", line 54, in extract_feat
x = self.backbone(img)
File "/home/syy/anaconda3/envs/fgd/lib/python3.7/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/syy/MixD/mmdet/models/backbones/resnet.py", line 627, in forward
x = self.conv1(x)
File "/home/syy/anaconda3/envs/fgd/lib/python3.7/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/syy/anaconda3/envs/fgd/lib/python3.7/site-packages/torch/nn/modules/conv.py", line 423, in forward
return self._conv_forward(input, self.weight)
File "/home/syy/anaconda3/envs/fgd/lib/python3.7/site-packages/torch/nn/modules/conv.py", line 420, in _conv_forward
self.padding, self.dilation, self.groups)
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
pppppM
