Skip to content
GitLab

Cross validation

Merged Anjith GEORGE requested to merge cross_validation into master

This MR adds the following

  • Adds cross-validation using a validation set; the best model based on the lowest validation loss is saved automatically now
  • Adds trainer, extractor, training script and unit tests for FASNet architecture
  • Small change in MCCNN architecture; added a flag for selecting whether to use sigmoid in eval phase.
Edited by Anjith GEORGE