Skip to content

Support for FLOPs in nn.Modules with input other than tensor

Choose a tag to compare

View all tags
@LukasHedegaard LukasHedegaard released this 17 Feb 13:21
· 28 commits to main since this release
0.3.1
fe57dff
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.

### Removed

  • try_custom_warmup.

Added

  • warm_up_fn overload option.
  • Support for FLOPs count in torch.nn.Module with input other than Tensor.
Assets 2
Loading