mmclassification/mmcls
fangxu 0aa789f3c3
[Feature] Add the frozen function for Swin Transformer model. (#574)
* Add the frozen function for Swin Transformer model

* add frozen parameter for swin transformer model

* add norm_eval parameter

* Delete =11.1

* Delete =418,driver

* delete _BatchNorm

* remove LayerNorm , add _BatchNorm

* unifying the style of frozen function refer ResNet

* Improve docs and add unit tests.

Co-authored-by: cxiang26 <cq.xiang@foxmail.com>
Co-authored-by: mzr1996 <mzr1996@163.com>
2021-12-07 11:58:14 +08:00
..
apis [Enchance] Set a random seed when the user does not set a seed. (#554) 2021-12-02 18:09:55 +08:00
core
datasets [Docs] Fix docs build dependency. (#584) 2021-12-07 11:16:33 +08:00
models [Feature] Add the frozen function for Swin Transformer model. (#574) 2021-12-07 11:58:14 +08:00
utils
__init__.py
version.py