Add --test option to specify test to run

pytorch / torchtitan

A native PyTorch Library for large model training

BSD 3-Clause "New" or "Revised" License

1.28k stars 115 forks source link

Closed kwen2501 closed 1 month ago

kwen2501 commented 1 month ago

Usage: --test <test_id>

Acceptable values: test_id in build_test_list (default: all)

Example:

rm -rf outputs && python test_runner.py outputs --test pp_gpipe