Skip to content

PyTorch Toolbelt 0.5.1
Compare
Choose a tag to compare
@BloodAxe BloodAxe released this 27 Jun 19:58
· 375 commits to master since this release
0.5.1
ee72463
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Verified
Learn about vigilant mode.

New API

  • Added fs.find_subdirectories_in_dir to retrieve list of subdirectories (non-recursive) in the given directory.
  • Added logodd averaging of TTA predictions and counterpart logodd_mean function.

Improvements

  • In plot_confusion_matrix one can disable plotting scores in each cell using show_scores argument (True by default).
  • freeze_model method now returns input module argument.
Assets 2
Loading