Skip to content

WIP: Generic trainer

  • 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

Loading