search

knsiczarnamagia / face-expression-recognizer

Face Expression Recognizer is a human face detector and face expression classificator. Created by members of the KNSI Czarna Magia AI Student Club.
GNU General Public License v3.0
4 stars 0 forks source link

Added trainer with callbacks #13

Closed Kiwinicki closed 6 months ago

Kiwinicki commented 6 months ago

I added Trainer class with some callbacks and hooks, but hooks massively slows down training on GPU and current implementation still lacks wandb integration. Everything works, but I still need to rewrite it to log activation stats to wandb.