I was trying to make a UNet-like architecture using PyTorch. I got a Runtime Error saying that there is not enough memory as I’m trying to allocate 103GB.
I am using a CPU only. And I have shrunk my input to [2,2,2] with a the batch_size=1, with no change to the error message.
Can anyone help me figure out why this is happening and how to fix it?