memory error

[lorenzori]

hi, thanks for the script. I am trying to run the same net on 400x400 images, however can't have a batch_size larger than 8 otherwise I get a memory error from my GPU (NVIDIA GRID K520). What hardware are you using for this?

[jocicmarko]

I'm using Titan GPU. Yes, these kind of models can be memory heavy, especially if you're using bigger input sizes. There are not much workarounds for this except:

• reducing input image size
• reducing batch size
• try to reduce your model memory footprint by using gradient checkpointing - https://github.com/openai/gradient-checkpointing