In this work, we apply techniques from the field of behavioral analysis and ethology to understand the learning dynamics of neural networks.
From conda:
First, make your conda environment:
conda env create -f environment.yml
Then, activate the environment:
conda activate edm_nn
Train:
# make sure you are in the root directory of the project
cd src
python train_network.py --config experiments/train_mnist.yaml
Replay:
cd src
python replay_network.py --config experiments/train_mnist.yaml --run_name mnist_mlp_20230531-201430
Analysis: Check the notebooks in src/notebooks