Grokking

How to use

Install dependencies

1. Install torch from its website
2. Install the rest:

conda create -n grokking python=3.8
conda activate grokking
conda install pytorch torchvision torchaudio -c pytorch
pip install -r src/requirement.txt

3. Configure Comet ML (Optional)

ipython

Run the following:

import comet_ml
comet_ml.init()

Train and Evaluation

python src/main.py --configs "configs/power_et_al_sum.conf" train