Skip to content

WIP: Generic trainer

Anjith GEORGE requested to merge generic_trainer into master
  • Adds a generic trainer class and trainer script
  • Moved the model specific stuff to config files; example config added
  • uses .to(device) throughout

Merge request reports