├── .gitignore ├── LICENSE ├── README.md ├── assets ├── grokfast_ma_loss_main.png ├── grokfast_ma_main.png ├── param_space.gif └── schematics.png ├── grokfast.py ├── main.py ├── main_imdb.py ├── main_mnist.py ├── main_qm9.py ├── requirements.txt └── results └── results_are_saved_here.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ironjr/grokfast/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ironjr/grokfast/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ironjr/grokfast/HEAD/README.md -------------------------------------------------------------------------------- /assets/grokfast_ma_loss_main.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ironjr/grokfast/HEAD/assets/grokfast_ma_loss_main.png -------------------------------------------------------------------------------- /assets/grokfast_ma_main.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ironjr/grokfast/HEAD/assets/grokfast_ma_main.png -------------------------------------------------------------------------------- /assets/param_space.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ironjr/grokfast/HEAD/assets/param_space.gif -------------------------------------------------------------------------------- /assets/schematics.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ironjr/grokfast/HEAD/assets/schematics.png -------------------------------------------------------------------------------- /grokfast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ironjr/grokfast/HEAD/grokfast.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ironjr/grokfast/HEAD/main.py -------------------------------------------------------------------------------- /main_imdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ironjr/grokfast/HEAD/main_imdb.py -------------------------------------------------------------------------------- /main_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ironjr/grokfast/HEAD/main_mnist.py -------------------------------------------------------------------------------- /main_qm9.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ironjr/grokfast/HEAD/main_qm9.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ironjr/grokfast/HEAD/requirements.txt -------------------------------------------------------------------------------- /results/results_are_saved_here.txt: -------------------------------------------------------------------------------- 1 | --------------------------------------------------------------------------------