v2.2.1 - Release 2.2.1 (#182)
🌟 Summary
THOP 2.2.1 improves operation-count estimates for models with adaptive max pooling or bilinear layers, without changing reported counts for Ultralytics’ buildable model configurations.
📊 Key Changes
- Fixed stride-based profiling: Adaptive max pools and
nn.Bilinearare now treated as fixed-size operations during extrapolation. - Released as
ultralytics-thop2.2.1: Publishes the profiling fix from PR #181. - Validated existing model counts: Ultralytics model FLOPs remained bit-identical across all buildable configuration files.
🎯 Purpose & Impact
- 📏 Prevents FLOPs from being overcounted when profiling models with these layers.
- 🔍 Makes complexity estimates more reliable for users comparing or evaluating affected models.
- ✅ Does not change model architectures or the reported FLOPs for the tested Ultralytics configurations.
What's Changed
- Refresh AGENTS.md and fix stride extrapolation after adaptive max pools by @glenn-jocher in #181
- Release 2.2.1 by @glenn-jocher in #182
Full Changelog: v2.2.0...v2.2.1