Hello,
I’m running my code on Tesla P100 (16 Go). l get stuck at CUDA out of memory, l get the following error after 17 epochs of training
line 2234, in forward
    x = nn.AvgPool1d(90, stride=None)(x)
  File "/local/anaconda3/envs/torch_edward/lib/python3.7/site-packages/torch/nn/modules/module.py", line 489, in __call__
    result = self.forward(*input, **kwargs)
  File "/local/anaconda3/envs/torch_edward/lib/python3.7/site-packages/torch/nn/modules/pooling.py", line 499, in forward
    self.count_include_pad)
RuntimeError: CUDA out of memory. Tried to allocate 1.12 MiB (GPU 0; 11.91 GiB total capacity; 5.52 GiB already allocated; 2.06 MiB free; 184.00 KiB cached)

The error comes from forward function :   " x = nn.AvgPool1d(90, stride=None)(x)"
def forward(): 
   x=self.layer1(x)
   x=self.layer2(x)
  .
  .
  .
  x=self.layer10(x)


  x=nn.AvgPool1d(90,stride=None)(x) # here l get the error

  x=x.squeeze(2)

  x=self.fc1(x)
  return x

Any cue ?
Thank you