Releases
v0.5.3
Compare
Sorry, something went wrong.
No results found
Talmaj
released this
04 Nov 20:44
Release Summary
BatchNorm Fixes & Improvements
Fixed critical bias bug : Corrected line 59 in batchnorm.py where bias was incorrectly set to scale instead of B
Fixed inference mode for batch_size > 1 : Explicitly set BatchNorm to eval mode to use running statistics (ONNX inference behavior) #44
Removed experimental flag : BatchNorm now works correctly with batch_size > 1 by default #35
Added comprehensive tests : Validated against onnxruntime with various batch sizes (1, 2, 4, 8), channels, spatial dimensions, epsilon values, and momentum values
ReduceSumSquare
Implemented ReduceSumSquare operator : Computes sum(x^2) along specified axes
Supports both opset versions : Handles axes as attribute (opset < 13) and as optional input (opset >= 13)
Comprehensive test coverage : 16+ parametrized test cases validating against onnxruntime with different input shapes, axes, and keepdims settings
LogSoftmax
Added LogSoftmax operator : Supports axis attribute with proper default handling (dim=-1) #49 #30
Added comprehensive tests : Each operator validated against onnxruntime with various input shapes, axes, and mathematical property verification
You can’t perform that action at this time.