PyTorch-Encoding icon indicating copy to clipboard operation
PyTorch-Encoding copied to clipboard

how to freeze syncbn?

Open xLuge opened this issue 4 years ago • 3 comments

hi, when i finetune the model,i need to freeze the syncbn. how can i do that?

xLuge avatar May 12 '21 06:05 xLuge

how r u fine tuning the model

ravitejarj avatar May 19 '21 08:05 ravitejarj

hi, when i finetune the model,i need to freeze the syncbn. how can i do that?

You may use the eval mode. model.eval()

zhanghang1989 avatar May 19 '21 17:05 zhanghang1989

how r u fine tuning the model i just fine tuning model like that https://hangzhang.org/PyTorch-Encoding/model_zoo/segmentation.html#pascal-voc-dataset but according to the deeplab3, we should freeze bn after coco pretrain.

xLuge avatar May 26 '21 07:05 xLuge