mmpretrain/tools
Wangbo Zhao(黑色枷锁) 14dcb69092
[Feature] Add mixmim backbone with checkpoints. (#1224)
* add mixmim backbone

* add mixmim inference

* add docstring, metafile, test and modify readme

* Update README and metafile

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-12-20 16:52:54 +08:00
..
analysis_tools [Docs] Improve the ResNet model page. (#1118) 2022-11-08 11:09:51 +08:00
misc [Docs] update visualization doc. (#1160) 2022-11-18 12:44:55 +08:00
model_converters [Feature] Add mixmim backbone with checkpoints. (#1224) 2022-12-20 16:52:54 +08:00
torchserve [Fix] Fix the torchserve. (#1143) 2022-11-21 11:18:42 +08:00
visualizations [Fix] Fix CAM visualization. (#1248) 2022-12-19 13:54:52 +08:00
dist_test.sh [Feature] Add multi machine dist_train (#734) 2022-03-23 12:03:03 +08:00
dist_train.sh [Feature] Add multi machine dist_train (#734) 2022-03-23 12:03:03 +08:00
kfold-cross-valid.py [Imporve] Update tools to enable `pin_memory` and `persistent_workers` by default. (#1024) 2022-09-14 11:57:32 +08:00
slurm_test.sh init commit 2020-05-21 21:21:43 +08:00
slurm_train.sh init commit 2020-05-21 21:21:43 +08:00
test.py [Imporve] Update tools to enable `pin_memory` and `persistent_workers` by default. (#1024) 2022-09-14 11:57:32 +08:00
train.py [Docs] Add runtime configuration docs. (#1128) 2022-11-02 10:59:59 +08:00