You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/test_engine
Jiarui Fang 4165eabb1e
[hotfix] remove potiential circle import (#1307)
2 years ago
..
test_engine.py [test] refactored with the new rerun decorator (#763) 3 years ago
test_gradient_accumluation.py [refactor] moving grad acc logic to engine (#804) 3 years ago