README.md

AffectNet Baseline using Pytorch Lightning

Cong M. Vo

Installation

pip install -r requirements.txt

Training

1. Train baseline

python train.py -c experiments/baseline.yaml --gpus 1 --num_workers 6

2. Train with Vit

python train.py -c experiments/vit.yaml --gpus 1 --num_workers 6