Implement correctness assertions for test_ema in Adam optimizer (#22979)
- Updated test_ema in keras/src/optimizers/adam_test.py with assertions for initialization, manual updates, and model.fit() integration. - Added test_ema_overwrite_frequency to verify periodic model weight overwriting with EMA values. - Verified changes with ruff and py_compile. Co-authored-by: andersendsa <199610634+andersendsa@users.noreply.github.com>
A
Andersen Dsa committed
36ef6cb522492e013ddd2ee472b1c16656e86a02
Parent: 60b1478
Committed by GitHub <noreply@github.com>
on 5/27/2026, 8:39:54 PM