A simple bootstrap for training custom models on AffectNet dataset on a system with multiple GPUs using Mirrored Distributed Strategy.
For downloading AffectNet, only Lab Managers, Professors or Students using their academic email account should fill out the request form HERE.
Dataset must be inside data
folder of the project.
For adding custom keras models, add a new python file inside models
folder and use it inside main.py
.
Keras callbacks will save weights and logs inside checkpoints
folder.