├── EVAL.md ├── LICENSE ├── README.md ├── TRAINING.md ├── _ET_pp_main_convnextB.py ├── _ET_pp_main_convnextS.py ├── _ET_pp_main_convnextT.py ├── _ET_pp_main_cswinB.py ├── _ET_pp_main_cswinS.py ├── _ET_pp_main_cswinT.py ├── _ET_pp_main_deitS.py ├── _ET_pp_main_res50.py ├── _ET_pp_main_swinB.py ├── _ET_pp_main_swinS.py ├── _ET_pp_main_swinT.py ├── _ET_pp_wReplay_main_cswinT.py ├── _ET_pp_wReplay_main_deitS.py ├── _ET_pp_wReplay_main_res50.py ├── _ET_pp_wReplay_main_swinT.py ├── datasets_buffer.py ├── engine_buffer.py ├── imgs ├── 300ep.png ├── coco.png ├── in_1k.png ├── in_22k.png ├── mae.png ├── overview.png ├── seg.png └── vary_epoch.png ├── lamb.py ├── main_buffer.py ├── models ├── convnext.py ├── convnext_isotropic.py ├── cswin.py ├── deit.py ├── metaformer.py └── swin.py ├── optim_factory.py └── utils_buffer.py /EVAL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/EVAL.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/README.md -------------------------------------------------------------------------------- /TRAINING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/TRAINING.md -------------------------------------------------------------------------------- /_ET_pp_main_convnextB.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/_ET_pp_main_convnextB.py -------------------------------------------------------------------------------- /_ET_pp_main_convnextS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/_ET_pp_main_convnextS.py -------------------------------------------------------------------------------- /_ET_pp_main_convnextT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/_ET_pp_main_convnextT.py -------------------------------------------------------------------------------- /_ET_pp_main_cswinB.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/_ET_pp_main_cswinB.py -------------------------------------------------------------------------------- /_ET_pp_main_cswinS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/_ET_pp_main_cswinS.py -------------------------------------------------------------------------------- /_ET_pp_main_cswinT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/_ET_pp_main_cswinT.py -------------------------------------------------------------------------------- /_ET_pp_main_deitS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/_ET_pp_main_deitS.py -------------------------------------------------------------------------------- /_ET_pp_main_res50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/_ET_pp_main_res50.py -------------------------------------------------------------------------------- /_ET_pp_main_swinB.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/_ET_pp_main_swinB.py -------------------------------------------------------------------------------- /_ET_pp_main_swinS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/_ET_pp_main_swinS.py -------------------------------------------------------------------------------- /_ET_pp_main_swinT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/_ET_pp_main_swinT.py -------------------------------------------------------------------------------- /_ET_pp_wReplay_main_cswinT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/_ET_pp_wReplay_main_cswinT.py -------------------------------------------------------------------------------- /_ET_pp_wReplay_main_deitS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/_ET_pp_wReplay_main_deitS.py -------------------------------------------------------------------------------- /_ET_pp_wReplay_main_res50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/_ET_pp_wReplay_main_res50.py -------------------------------------------------------------------------------- /_ET_pp_wReplay_main_swinT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/_ET_pp_wReplay_main_swinT.py -------------------------------------------------------------------------------- /datasets_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/datasets_buffer.py -------------------------------------------------------------------------------- /engine_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/engine_buffer.py -------------------------------------------------------------------------------- /imgs/300ep.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/imgs/300ep.png -------------------------------------------------------------------------------- /imgs/coco.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/imgs/coco.png -------------------------------------------------------------------------------- /imgs/in_1k.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/imgs/in_1k.png -------------------------------------------------------------------------------- /imgs/in_22k.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/imgs/in_22k.png -------------------------------------------------------------------------------- /imgs/mae.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/imgs/mae.png -------------------------------------------------------------------------------- /imgs/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/imgs/overview.png -------------------------------------------------------------------------------- /imgs/seg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/imgs/seg.png -------------------------------------------------------------------------------- /imgs/vary_epoch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/imgs/vary_epoch.png -------------------------------------------------------------------------------- /lamb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/lamb.py -------------------------------------------------------------------------------- /main_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/main_buffer.py -------------------------------------------------------------------------------- /models/convnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/models/convnext.py -------------------------------------------------------------------------------- /models/convnext_isotropic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/models/convnext_isotropic.py -------------------------------------------------------------------------------- /models/cswin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/models/cswin.py -------------------------------------------------------------------------------- /models/deit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/models/deit.py -------------------------------------------------------------------------------- /models/metaformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/models/metaformer.py -------------------------------------------------------------------------------- /models/swin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/models/swin.py -------------------------------------------------------------------------------- /optim_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/optim_factory.py -------------------------------------------------------------------------------- /utils_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/EfficientTrain/HEAD/utils_buffer.py --------------------------------------------------------------------------------