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