Traceback (most recent call last):


 "test.py", line 114, in <module>

args.scale).cpu() + data['bicubic']


 "/home/rahid/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__

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


 "/home/rahid/proSR/lib/prosr/models/generators.py", line 124, in forward

feats = getattr(self, 'pyramid_residual_%d' % s)(feats) + feats


 "/home/rahid/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__

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


 "/home/rahid/.local/lib/python3.6/site-packages/torch/nn/modules/container.py", line 91, in forward

input = module(input)


 "/home/rahid/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__

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


 "/home/rahid/proSR/lib/prosr/models/layers.py", line 179, in forward

self).forward(x) + identity_x


 "/home/rahid/.local/lib/python3.6/site-packages/torch/nn/modules/container.py", line 91, in forward

input = module(input)


 "/home/rahid/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__

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


 "/home/rahid/.local/lib/python3.6/site-packages/torch/nn/modules/container.py", line 91, in forward

input = module(input)


 "/home/rahid/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__

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


 "/home/rahid/proSR/lib/prosr/models/layers.py", line 150, in forward

return torch.cat([x, new_features], 1)
RuntimeError: CUDA error: out of memory

Traceback (most recent call last):


 "/usr/local/lib/python3.7/tkinter/__init__.py", line 3504, in __del__

self.tk.call('image', 'delete', self.name)
RuntimeError: main thread is not in main loop
