Oneflow-Inc / OneAutoTest

Auto-Test System
Apache License 2.0
5 stars 6 forks source link

add arg_train_platform.sh #87

Closed Tendo33 closed 8 months ago

Tendo33 commented 1 year ago

因为三平台的训练脚本,只有环境变量的区别,所以在输入参数加了一个PLATFORM参数,用0,1,2,3来表示各平台来使用各平台的环境变量,这样可以直接统一训练语句,使用一个训练脚本(run_gpt.sh)就可以在三平台运行,只需要更改PLATFORM参数就可以了。