Skip to content

Fix PyTorch hooks compatibility
Compare
Choose a tag to compare
@LukasHedegaard LukasHedegaard released this 06 Sep 12:22
· 305 commits to main since this release
v0.11.1
8b0ea68
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

Added

  • Add call_mode.
  • Add warm_up.

Changed

  • Container implementations to use __call__ with alternating call_modes. This change was necessary to propperly trigger the torch hooks needed in ptflops.

Fixed

  • ptflops compatibility.
Assets 2