├── .gitignore ├── README.md ├── figures ├── acc.png └── loss.png ├── requirements.txt └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/teddykoker/grokking/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/teddykoker/grokking/HEAD/README.md -------------------------------------------------------------------------------- /figures/acc.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/teddykoker/grokking/HEAD/figures/acc.png -------------------------------------------------------------------------------- /figures/loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/teddykoker/grokking/HEAD/figures/loss.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | matplotlib==3.4.3 2 | torch==1.10.0 3 | tqdm==4.62.3 4 | -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/teddykoker/grokking/HEAD/train.py --------------------------------------------------------------------------------