Traceback (most recent call last):


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

results = model.detect([image])


 "/home/qq4060/pytorch-mask-rcnn/model.py", line 1598, in detect

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


 "/home/qq4060/pytorch-mask-rcnn/model.py", line 1645, in predict

layer_outputs.append(self.rpn(p))


 "/home/qq4060/.local/lib/python3.5/site-packages/torch/nn/modules/module.py", line 491, in __call__

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


 "/home/qq4060/pytorch-mask-rcnn/model.py", line 879, in forward

x = self.relu(self.conv_shared(self.padding(x)))


 "/home/qq4060/.local/lib/python3.5/site-packages/torch/nn/modules/module.py", line 491, in __call__

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


 "/home/qq4060/.local/lib/python3.5/site-packages/torch/nn/modules/conv.py", line 301, in forward

self.padding, self.dilation, self.groups)
RuntimeError: cuda runtime error (2) : out of memory at /pytorch/aten/src/THC/generic/THCStorage.cu:58

Traceback (most recent call last):


 "/home/xxx/Codes/check_stanza.py", line 5, in <module>

nlp=stanza.Pipeline(lang='en', processors='tokenize',use_gpu=True)


 "/home/xxx/anaconda3/envs/pt/lib/python3.6/site-packages/stanza/pipeline/core.py", line 120, in __init__

use_gpu=self.use_gpu)


 "/home/xxx/anaconda3/envs/pt/lib/python3.6/site-packages/stanza/pipeline/processor.py", line 103, in __init__

self._set_up_model(config, use_gpu)


 "/home/xxx/anaconda3/envs/pt/lib/python3.6/site-packages/stanza/pipeline/tokenize_processor.py", line 38, in _set_up_model

self._trainer = Trainer(model_file=config['model_path'], use_cuda=use_gpu)


 "/home/xxx/anaconda3/envs/pt/lib/python3.6/site-packages/stanza/models/tokenize/trainer.py", line 27, in __init__

self.model.cuda()


 "/home/xxx/anaconda3/envs/pt/lib/python3.6/site-packages/torch/nn/modules/module.py", line 260, in cuda

return self._apply(lambda t: t.cuda(device))


 "/home/xxx/anaconda3/envs/pt/lib/python3.6/site-packages/torch/nn/modules/module.py", line 187, in _apply

module._apply(fn)


 "/home/xxx/anaconda3/envs/pt/lib/python3.6/site-packages/torch/nn/modules/module.py", line 193, in _apply

param.data = fn(param.data)


 "/home/xxx/anaconda3/envs/pt/lib/python3.6/site-packages/torch/nn/modules/module.py", line 260, in <lambda>

return self._apply(lambda t: t.cuda(device))
RuntimeError: CUDA error: out of memory
