To use this repo, create a conda environment using environment.yml or requirements.txt
# from environment.yml (recommended)
conda env create -f environment.yml
# from requirements.txt
conda create --name <env> --file requirements.txt
Download the offical fer2013 dataset, and place it in the outmost folder with the following folder structure datasets/fer2013/fer2013.csv
To train your own version of our network, run the following
python train.py network=vgg name=my_vgg
To change the default parameters, you may also add arguments such as bs=128 or lr=0.1. For more details, please refer to utils/hparams.py