Mask3D
Mask3D copied to clipboard
JonasSchult
Batch size selection
Hello, thank you for the excellent work! I'm trying to train on smaller batches, so I can maintain the same resolution but still have the same results on a weaker GPU. Is there any parameter to change the batch size? Thank you in advance, Jonas.
Salutations, Guilherme Coelho Fermino.
Found it! However, now I'm a bit worried about the learning rate, which its implementation on Mask3D is inverselly proportional to the batch size, and I couldn't find a parameter to change that. Shouldn't it be direcly proportional do the batch size? (as I've only used softgroup, maybe I could be mistaken with the implementation of Mask3D)
Dear sir,I have the same issue, can you tell me where to change the batch_size? Thank you very much!
The parameter inputed is data.batch_size, then you input the value you want or your machine can handle. Like:
python main_instance_segmentation.py \
data.batch_size=1
@guilherme-coelho-isi Hi! did u address the learning rate issue above? And did you try the training with 1 batch size? How is the performance compared to the original Mask3D?
