What to change to increase model capacity?

[Tommaso Bendinelli]

Hello,

I have an extremely hard segmentation task. The network (SOLOv2) is struggling a lot even in the training set. What parameters should I change to increase model learning capacity? Does increasing the grid size help?

Best,
Tommaso