Traceback (most recent call last):
File “refcoco/train_end2end.py”, line 60, in
main()
File “refcoco/train_end2end.py”, line 54, in main
rank, model = train_net(args, config)
File “/content/gdrive/My Drive/DDP/VL-BERT/refcoco/…/refcoco/function/train.py”, line 323, in train_net
gradient_accumulate_steps=config.TRAIN.GRAD_ACCUMULATE_STEPS)
File “/content/gdrive/My Drive/DDP/VL-BERT/refcoco/…/common/trainer.py”, line 115, in train
outputs, loss = net(*batch)
File “/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py”, line 727, in _call_impl
result = self.forward(*input, **kwargs)
File “/content/gdrive/My Drive/DDP/VL-BERT/refcoco/…/common/module.py”, line 22, in forward
return self.train_forward(*inputs, **kwargs)
File “/content/gdrive/My Drive/DDP/VL-BERT/refcoco/…/refcoco/modules/resnet_vlbert_for_refcoco.py”, line 96, in train_forward
segms=None)
File “/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py”, line 727, in _call_impl
result = self.forward(*input, **kwargs)
File “/content/gdrive/My Drive/DDP/VL-BERT/refcoco/…/common/fast_rcnn.py”, line 149, in forward
roi_align_res = self.roi_align(img_feats[‘body4’], rois).type(images.dtype)
File “/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py”, line 727, in _call_impl
result = self.forward(*input, **kwargs)
File “/content/gdrive/My Drive/DDP/VL-BERT/refcoco/…/common/lib/roi_pooling/roi_align.py”, line 69, in forward
input.float(), rois.float(), self.output_size, self.spatial_scale, self.sampling_ratio
File “/content/gdrive/My Drive/DDP/VL-BERT/refcoco/…/common/lib/roi_pooling/roi_align.py”, line 20, in forward
input, rois, spatial_scale, output_size[0], output_size[1], sampling_ratio
RuntimeError: Unrecognized tensor type ID: AutogradCUDA
I am running on google colab, using pytorch version 1.7.0 , torchvision 0.8.1 and cuda 10.1. Same error is coming with cuda 9.2 also.