hanhaowen-mt 3d8a611eec

[Feature] Add the support for musa device support (#1453)

2024-01-11 16:25:01 +08:00

..

test_activation_checkpointing.py

[Feature] Implement gradient checkpointing (#1319)

2023-09-04 23:29:24 +08:00

test_amp.py

[Feature] Add the support for musa device support (#1453)

2024-01-11 16:25:01 +08:00

test_checkpoint.py

…

test_log_processor.py

[Feature] Add the support for musa device support (#1453)

2024-01-11 16:25:01 +08:00

test_priority.py

…

test_runner.py

[Feature] Runner supports setting the number of iterations for per epoch (#1292)

2023-10-08 16:43:44 +08:00