├── .gitignore ├── LICENSE ├── README.md ├── classification ├── README.md ├── cifar.py └── imagenet.py ├── docs └── images │ ├── cifar10_wd.png │ ├── classify_full.png │ └── language_model.png ├── language_model ├── README.md ├── model_word_ada │ ├── __init__.py │ ├── adaptive.py │ ├── dataset.py │ └── lm.py ├── pre_word_ada │ └── encode_data2folder.py └── train_1bw.py ├── optim ├── __init__.py ├── adabelief.py ├── adahessian.py ├── apollo.py ├── lr_scheduler.py └── radamw.py └── utils ├── __init__.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XuezheMax/apollo/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XuezheMax/apollo/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XuezheMax/apollo/HEAD/README.md -------------------------------------------------------------------------------- /classification/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XuezheMax/apollo/HEAD/classification/README.md -------------------------------------------------------------------------------- /classification/cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XuezheMax/apollo/HEAD/classification/cifar.py -------------------------------------------------------------------------------- /classification/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XuezheMax/apollo/HEAD/classification/imagenet.py -------------------------------------------------------------------------------- /docs/images/cifar10_wd.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XuezheMax/apollo/HEAD/docs/images/cifar10_wd.png -------------------------------------------------------------------------------- /docs/images/classify_full.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XuezheMax/apollo/HEAD/docs/images/classify_full.png -------------------------------------------------------------------------------- /docs/images/language_model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XuezheMax/apollo/HEAD/docs/images/language_model.png -------------------------------------------------------------------------------- /language_model/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XuezheMax/apollo/HEAD/language_model/README.md -------------------------------------------------------------------------------- /language_model/model_word_ada/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /language_model/model_word_ada/adaptive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XuezheMax/apollo/HEAD/language_model/model_word_ada/adaptive.py -------------------------------------------------------------------------------- /language_model/model_word_ada/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XuezheMax/apollo/HEAD/language_model/model_word_ada/dataset.py -------------------------------------------------------------------------------- /language_model/model_word_ada/lm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XuezheMax/apollo/HEAD/language_model/model_word_ada/lm.py -------------------------------------------------------------------------------- /language_model/pre_word_ada/encode_data2folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XuezheMax/apollo/HEAD/language_model/pre_word_ada/encode_data2folder.py -------------------------------------------------------------------------------- /language_model/train_1bw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XuezheMax/apollo/HEAD/language_model/train_1bw.py -------------------------------------------------------------------------------- /optim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XuezheMax/apollo/HEAD/optim/__init__.py -------------------------------------------------------------------------------- /optim/adabelief.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XuezheMax/apollo/HEAD/optim/adabelief.py -------------------------------------------------------------------------------- /optim/adahessian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XuezheMax/apollo/HEAD/optim/adahessian.py -------------------------------------------------------------------------------- /optim/apollo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XuezheMax/apollo/HEAD/optim/apollo.py -------------------------------------------------------------------------------- /optim/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XuezheMax/apollo/HEAD/optim/lr_scheduler.py -------------------------------------------------------------------------------- /optim/radamw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XuezheMax/apollo/HEAD/optim/radamw.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XuezheMax/apollo/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XuezheMax/apollo/HEAD/utils/utils.py --------------------------------------------------------------------------------