Traceback (most recent call last):


 "onnx_export.py", line 34, in <module>

_ = model(img)
# dry run


 "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 550, in __call__

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


 "/content/yolov5/models/yolo.py", line 90, in forward

return self.forward_once(x, profile)
# single-scale inference, train


 "/content/yolov5/models/yolo.py", line 110, in forward_once

x = m(x)
# run


 "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 550, in __call__

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


 "/content/yolov5/models/common.py", line 87, in forward

return self.conv(torch.cat([x[..., ::2, ::2], x[..., 1::2, ::2], x[..., ::2, 1::2], x[..., 1::2, 1::2]], 1))


 "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 550, in __call__

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


 "/content/yolov5/models/common.py", line 25, in fuseforward

return self.act(self.conv(x))


 "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 550, in __call__

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


 "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/conv.py", line 353, in forward

return self._conv_forward(input, self.weight)


 "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/conv.py", line 350, in _conv_forward

self.padding, self.dilation, self.groups)
RuntimeError: Expected 4-dimensional input for 4-dimensional weight [32, 12, 3, 3], but got 3-dimensional input of size [1, 6, 208] instead

Original Traceback (most recent call last):


 "/home/ubuntu/miniconda3/envs/py37/lib/python3.7/site-packages/torch/nn/parallel/parallel_apply.py", line 60, in _worker

output = module(*input, **kwargs)


 "/home/ubuntu/miniconda3/envs/py37/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in __call__

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


 "/home/ubuntu/miniconda3/envs/py37/lib/python3.7/site-packages/apex/amp/_initialize.py", line 197, in new_fwd

**applier(kwargs, input_caster))


 "/home/ubuntu/pycharm/FARM/farm/modeling/adaptive_model.py", line 222, in forward

**kwargs, output_all_encoded_layers=False


 "/home/ubuntu/miniconda3/envs/py37/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in __call__

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


 "/home/ubuntu/pycharm/FARM/farm/modeling/language_model.py", line 355, in forward

attention_mask=padding_mask,


 "/home/ubuntu/miniconda3/envs/py37/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in __call__

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


 "/home/ubuntu/miniconda3/envs/py37/lib/python3.7/site-packages/transformers/modeling_bert.py", line 735, in forward

embedding_output = self.embeddings(input_ids=input_ids, position_ids=position_ids, token_type_ids=token_type_ids, inputs_embeds=inputs_embeds)


 "/home/ubuntu/miniconda3/envs/py37/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in __call__

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


 "/home/ubuntu/miniconda3/envs/py37/lib/python3.7/site-packages/transformers/modeling_bert.py", line 186, in forward

inputs_embeds = self.word_embeddings(input_ids)


 "/home/ubuntu/miniconda3/envs/py37/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in __call__

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


 "/home/ubuntu/miniconda3/envs/py37/lib/python3.7/site-packages/torch/nn/modules/sparse.py", line 114, in forward

self.norm_type, self.scale_grad_by_freq, self.sparse)


 "/home/ubuntu/miniconda3/envs/py37/lib/python3.7/site-packages/torch/nn/functional.py", line 1484, in embedding

return torch.embedding(weight, input, padding_idx, scale_grad_by_freq, sparse)
RuntimeError: arguments are located on different GPUs at /opt/conda/conda-bld/pytorch_1579022060824/work/aten/src/THC/generic/THCTensorIndex.cu:400
