Yash_Belhe:

max_pool2d_with_indices_out_cuda_frame


This is the error that I’m getting. Here’s a full version of the error
Traceback (most recent call last):
  File "train_new.py", line 197, in <module>
    train_net(net=net,
  File "train_new.py", line 90, in train_net
    pred_rgb = net(bayer_input)
  File "/home/ybelhe/anaconda3/envs/pytorch_p3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 532, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/ybelhe/code/training/unet/unet_model.py", line 51, in forward
    x4 = self.down3(x3)
  File "/home/ybelhe/anaconda3/envs/pytorch_p3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 532, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/ybelhe/code/training/unet/unet_parts.py", line 39, in forward
    return self.maxpool_conv(x)
  File "/home/ybelhe/anaconda3/envs/pytorch_p3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 532, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/ybelhe/anaconda3/envs/pytorch_p3/lib/python3.8/site-packages/torch/nn/modules/container.py", line 100, in forward
    input = module(input)
  File "/home/ybelhe/anaconda3/envs/pytorch_p3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 532, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/ybelhe/anaconda3/envs/pytorch_p3/lib/python3.8/site-packages/torch/nn/modules/pooling.py", line 139, in forward
    return F.max_pool2d(input, self.kernel_size, self.stride,
  File "/home/ybelhe/anaconda3/envs/pytorch_p3/lib/python3.8/site-packages/torch/_jit_internal.py", line 181, in fn
    return if_false(*args, **kwargs)
  File "/home/ybelhe/anaconda3/envs/pytorch_p3/lib/python3.8/site-packages/torch/nn/functional.py", line 487, in _max_pool2d
    return torch.max_pool2d(
RuntimeError: max_pool2d_with_indices_out_cuda_frame failed with error code 0

Please do let me know if there’s any other information that I can provide you with