I have an image that I want to use for my transfer style, it is 512x320 in size and 93,7ko.
I downloaded the reference notebook through Snapchat.
I put my style and reference image in the notebook and I run all the code.
In the train loop, they show me this error, I don't understand.
RuntimeError Traceback (most recent call last)
<ipython-input-199-0b6d284ff067> in <module>() 13 noise = torch.zeros_like(image) 14 noise.normal_(mean=0, std=4) ---> 15 styled_image = model(image + noise) 16 17 features_x = vgg(image, features_num=2)
/usr/local/lib/python3.6/dist-packages/torch/nn/modules/conv.py in conv2d_forward(self, input, weight) 340 _pair(0), self.dilation, self.groups) 341 return F.conv2d(input, weight, self.bias, self.stride, --> 342 self.padding, self.dilation, self.groups) 343 344 def forward(self, input):
RuntimeError: CUDA out of memory. Tried to allocate 128.00 MiB (GPU 0; 11.17 GiB total capacity; 10.52 GiB already allocated; 110.69 MiB free; 10.75 GiB reserved in total by PyTorch)
I tried to reduce the file size but it doesn't change anything.
Is it still the size of the image that is too heavy?