chongruo / detectron2-ResNeSt

A fork of Detectron2 with ResNeSt backbone
https://arxiv.org/abs/2004.08955
Apache License 2.0
385 stars 73 forks source link

How to train with batch size 1 #60

Open ptran1203 opened 2 years ago

ptran1203 commented 2 years ago

I try to train an instance segmentation model with resnest200 backbone with image size 1024. But due to memory constraint, I can fit only batch size of 1. But when I use SyncBN, It requires more than 1 sample. Can I use other batch norm (or set norm to None) without effect performance