RuntimeError: cuda runtime error (2) : out of memory at /pytorch/aten/src/THC/generic/THCStorage.cu:58

5

1
11443×644 140 KB

problem is at loss.backward(), but I cannot figure out what’s wrong with my code.