diff --git a/examples/opensoraplan1.2/pretrain_opensoraplan1_2.sh b/examples/opensoraplan1.2/pretrain_opensoraplan1_2.sh index 2808460cb204e3ac9b90b2b85a13cc5063496f7b..e1b9164e6f11e15be920da625f3a4ace5e1bcf63 100644 --- a/examples/opensoraplan1.2/pretrain_opensoraplan1_2.sh +++ b/examples/opensoraplan1.2/pretrain_opensoraplan1_2.sh @@ -26,6 +26,7 @@ GBS=$(($WORLD_SIZE*$MBS/$CP/$TP/$PP*$ACC)) MM_DATA="./examples/opensoraplan1.2/data.json" MM_MODEL="./examples/opensoraplan1.2/model_opensoraplan1_2.json" MM_TOOL="./mindspeed_mm/tools/tools.json" +SAVE_PATH="your_ckpt_path_to_save" DISTRIBUTED_ARGS=" --nproc_per_node $GPUS_PER_NODE \