[Tests] Adds a test suite for `EMAModel` #2530
ema test cases.
31183ee1
debugging maessages.
313414ac
debugging maessages.
f47487fa
add: tests for ema.
2d0f7b57
fix: optimization_step arg,
17994e56
handle device placement.
49aed340
Apply suggestions from code review
258bbe11
remove del and gc.
f4c4e0c3
address PR feedback.
88502754
sayakpaul
marked this pull request as ready for review 2 years ago
add: tests for serialization.
d6241d20
fix: typos.
aee28461
Merge branch 'main' into tests/ema
b1a19d9d
Merge branch 'main' into tests/ema
0ead0380
skip_mps to serialization.
8f2a75f3
sayakpaul
merged
f9cfb5ab
into main 2 years ago
sayakpaul
deleted the tests/ema branch 2 years ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub