Traceback (most recent call last):


 "scripts/integration_test.py", line 115, in <module>

seq = predictor.predict(inp_seq.split())


 "build/bdist.linux-x86_64/egg/seq2seq/evaluator/predictor.py", line 44, in predict


 "build/bdist.linux-x86_64/egg/seq2seq/evaluator/predictor.py", line 30, in get_decoder_features


 "/root/deep_rl_regex/local/lib/python2.7/site-packages/torch/nn/modules/module.py", line 491, in __call__

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


 "build/bdist.linux-x86_64/egg/seq2seq/models/seq2seq.py", line 53, in forward


 "/root/deep_rl_regex/local/lib/python2.7/site-packages/torch/nn/modules/module.py", line 491, in __call__

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


 "build/bdist.linux-x86_64/egg/seq2seq/models/TopKDecoder.py", line 131, in forward


 "build/bdist.linux-x86_64/egg/seq2seq/models/DecoderRNN.py", line 96, in forward_step


 "/root/deep_rl_regex/local/lib/python2.7/site-packages/torch/nn/modules/module.py", line 491, in __call__

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


 "/root/deep_rl_regex/local/lib/python2.7/site-packages/torch/nn/modules/sparse.py", line 108, in forward

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


 "/root/deep_rl_regex/local/lib/python2.7/site-packages/torch/nn/functional.py", line 1076, in embedding

return torch.embedding(weight, input, padding_idx, scale_grad_by_freq, sparse)
RuntimeError: Expected object of type torch.cuda.LongTensor but found type torch.LongTensor for argument #3 'index'

Traceback (most recent call last):


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

results = model.detect([image])


 "/path-to/pytorch-mask-rcnn/model.py", line 1598, in detect

detections, mrcnn_mask = self.predict([molded_images, image_metas], mode='inference')


 "/path-to/pytorch-mask-rcnn/model.py", line 1636, in predict

[p2_out, p3_out, p4_out, p5_out, p6_out] = self.fpn(molded_images)


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

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


 "/path-to/pytorch-mask-rcnn/model.py", line 181, in forward

p2_out = self.P2_conv1(c2_out) + F.upsample(p3_out, scale_factor=2)
RuntimeError: cuda runtime error (2) : out of memory at /path-to/pytorch/aten/src/THC/generic/THCStorage.cu:58
