[go: up one dir, main page]
Skip to content

Usability improvements

Johann Benerradirequested to merge
HanBnrd/bnstealth:improvements into main

Changes

  • Enable loading on CPU a model trained on GPU and vice versa
  • Fix loading models trained on multiple GPUs
  • Avoid hyperparameter search when only one combination is provided
  • Enable more granularity in checkpoint loading to continue an incomplete training
  • Add loss graph plotting for each fold separately
  • Make directory creation for results storage more reliable
  • Make manual seed more reliable
  • Clean up imports in the documentation example

Merge request reports