Commit c4e8b04
authored
Fix custom classifier false positives by removing BatchNormalization layers (#824)
Resolves train-inference mismatch that caused ~100% confidence scores on
background noise in custom classifiers trained with v2.0.0+.
Root cause: BatchNormalization layers were added in v2.0.0 but received
different input distributions during training (L2-normalized embeddings)
vs inference (raw embeddings), breaking learned decision boundaries.
Solution: Remove BatchNormalization layers from build_linear_classifier()
to restore reliable v1.5.1 behavior while keeping other v2.x improvements
(L2 regularization, improved learning rate schedule).
Fixes #8231 parent a62d4ce commit c4e8b04
1 file changed
Lines changed: 0 additions & 6 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
635 | 635 | | |
636 | 636 | | |
637 | 637 | | |
638 | | - | |
639 | | - | |
640 | | - | |
641 | 638 | | |
642 | 639 | | |
643 | 640 | | |
| |||
650 | 647 | | |
651 | 648 | | |
652 | 649 | | |
653 | | - | |
654 | | - | |
655 | | - | |
656 | 650 | | |
657 | 651 | | |
658 | 652 | | |
| |||
0 commit comments