Wing Lian 6130e40c37 fix flaky tests; should be using train loss from final step rather than final avg train loss

2026-03-22 10:38:46 -04:00

..

__init__.py

don't run multigpu tests twice, run SP in separate test (#2542)

2025-04-21 10:24:13 -04:00

test_sp.py

fix flaky tests; should be using train loss from final step rather than final avg train loss

2026-03-22 10:38:46 -04:00