Traceback (most recent call last):


 "main.py", line 20, in <module>

t.train()


 "C:\Users\motor\RCAN-master\RCAN_TrainCode\code\trainer.py", line 51, in train

sr = self.model(lr, idx_scale)


 "C:\Users\motor\Anaconda3\envs\TENSORFLOW\lib\site-packages\torch\nn\modules\module.py", line 491, in __call__

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


 "C:\Users\motor\RCAN-master\RCAN_TrainCode\code\model\__init__.py", line 54, in forward

return self.model(x)


 "C:\Users\motor\Anaconda3\envs\TENSORFLOW\lib\site-packages\torch\nn\modules\module.py", line 491, in __call__

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


 "C:\Users\motor\RCAN-master\RCAN_TrainCode\code\model\rcan.py", line 110, in forward

res = self.body(x)


 "C:\Users\motor\Anaconda3\envs\TENSORFLOW\lib\site-packages\torch\nn\modules\module.py", line 491, in __call__

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


 "C:\Users\motor\Anaconda3\envs\TENSORFLOW\lib\site-packages\torch\nn\modules\container.py", line 91, in forward

input = module(input)


 "C:\Users\motor\Anaconda3\envs\TENSORFLOW\lib\site-packages\torch\nn\modules\module.py", line 491, in __call__

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


 "C:\Users\motor\RCAN-master\RCAN_TrainCode\code\model\rcan.py", line 62, in forward

res = self.body(x)


 "C:\Users\motor\Anaconda3\envs\TENSORFLOW\lib\site-packages\torch\nn\modules\module.py", line 491, in __call__

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


 "C:\Users\motor\Anaconda3\envs\TENSORFLOW\lib\site-packages\torch\nn\modules\container.py", line 91, in forward

input = module(input)


 "C:\Users\motor\Anaconda3\envs\TENSORFLOW\lib\site-packages\torch\nn\modules\module.py", line 491, in __call__

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


 "C:\Users\motor\RCAN-master\RCAN_TrainCode\code\model\rcan.py", line 44, in forward

res = self.body(x)


 "C:\Users\motor\Anaconda3\envs\TENSORFLOW\lib\site-packages\torch\nn\modules\module.py", line 491, in __call__

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


 "C:\Users\motor\Anaconda3\envs\TENSORFLOW\lib\site-packages\torch\nn\modules\container.py", line 91, in forward

input = module(input)


 "C:\Users\motor\Anaconda3\envs\TENSORFLOW\lib\site-packages\torch\nn\modules\module.py", line 491, in __call__

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


 "C:\Users\motor\RCAN-master\RCAN_TrainCode\code\model\rcan.py", line 25, in forward

return x * y
RuntimeError: cuda runtime error (2) : out of memory at c:\programdata\miniconda3\conda-bld\pytorch_1524543037166\work\aten\src\thc\generic/THCStorage.cu:58

Traceback (most recent call last):


 "tools/train.py", line 245, in <module>

main()


 "tools/train.py", line 241, in main

train()


 "tools/train.py", line 187, in train

loss_pre = criteria_pre(logits, lb)


 "/home/sjx/anaconda3/envs/py36bisenet/lib/python3.6/site-packages/torch/nn/modules/module.py", line 722, in _call_impl

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


 "./lib/ohem_ce_loss.py", line 38, in forward

loss_hard = loss[loss > self.thresh]
RuntimeError: copy_if failed to synchronize: cudaErrorIllegalAddress: an illegal memory access was encountered
