/AffectNet-Baseline

Primary LanguagePython

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
  1. Train with Vit
python train.py -c experiments/vit.yaml --gpus 1 --num_workers 6