search

zhanghang1989 / PyTorch-Encoding

A CV toolkit for my papers.
https://hangzhang.org/PyTorch-Encoding/
MIT License
2.04k stars 450 forks source link

how to freeze syncbn? #392

Open xLuge opened 3 years ago

xLuge commented 3 years ago

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

ravitejarj commented 3 years ago

how r u fine tuning the model

zhanghang1989 commented 3 years ago

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()

xLuge commented 3 years ago

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.