This code version is sooo ~ bad! and memory error why???

[ballade8]

i was very difficult to match code to right version of packages...

anyway

THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1535491974311/work/aten/src/THC/THCGeneral.cpp line=663 error=11 : invalid argument Traceback (most recent call last): File "/home/kaist/PycharmProjects/NTS-Net/train.py", line 66, in raw_logits, concat_logits, part_logits, _, top_n_prob = net(img) File "/home/kaist/anaconda3/envs/TASN-1/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call result = self.forward(*input, **kwargs) File "/home/kaist/anaconda3/envs/TASN-1/lib/python3.6/site-packages/torch/nn/parallel/data_parallel.py", line 121, in forward return self.module(*inputs[0], **kwargs[0]) File "/home/kaist/anaconda3/envs/TASN-1/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call result = self.forward(*input, **kwargs) File "/home/kaist/PycharmProjects/NTS-Net/core/model.py", line 68, in forward _, _, part_features = self.pretrained_model(part_imgs.detach()) File "/home/kaist/anaconda3/envs/TASN-1/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call result = self.forward(*input, **kwargs) File "/home/kaist/PycharmProjects/NTS-Net/core/resnet.py", line 142, in forward x = self.layer2(x) File "/home/kaist/anaconda3/envs/TASN-1/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call result = self.forward(*input, **kwargs) File "/home/kaist/anaconda3/envs/TASN-1/lib/python3.6/site-packages/torch/nn/modules/container.py", line 91, in forward input = module(input) File "/home/kaist/anaconda3/envs/TASN-1/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call result = self.forward(*input, **kwargs) File "/home/kaist/PycharmProjects/NTS-Net/core/resnet.py", line 74, in forward out = self.conv1(x) File "/home/kaist/anaconda3/envs/TASN-1/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call result = self.forward(*input, **kwargs) File "/home/kaist/anaconda3/envs/TASN-1/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 301, in forward self.padding, self.dilation, self.groups) RuntimeError: CUDA error: out of memory

Process finished with exit code 1