Fix error in freezing batch norm variables. Previously, only batch mean...
Fix error in freezing batch norm variables. Previously, only batch mean statistics were frozen, leading to poor accuracy. This fix freezes both batch mean and variance statistics PiperOrigin-RevId: 207625559
Loading
Please sign in to comment