torch.cuda.OutOfMemoryError when training Mask R-CNN

[ptrblck]

@antonl

which I don’t expect since the model is ~200MB and the batch is ~300MB.

The model parameter and input size might be tiny compared to the stored intermediates needed for gradient computation.
Have a look at this post showing an example.