forked from jiuyuan/CPM-9G-8B
Compare commits
2 Commits
8f38b2f740
...
acc0a1b717
Author | SHA1 | Date |
---|---|---|
wql | acc0a1b717 | |
wql | 092e08a149 |
|
@ -46,7 +46,7 @@ OPTS+=" --save-origin-model"
|
|||
|
||||
OPTS+=" $@"
|
||||
|
||||
CMD="torchrun --nnodes=1 --nproc_per_node=2 --rdzv_id=1 --rdzv_backend=c10d --rdzv_endpoint=${MASTER_ADDR}:${MASTER_PORT} ${CPM_PATH}/apps/cpm9g/sft_cpm9g_delta.py ${OPTS}"
|
||||
CMD="FORCE_TORCHRUN=1 CUDA_VISIBLE_DEVICES=2,3,4,5,6 torchrun --nnodes=1 --nproc_per_node=2 --rdzv_id=1 --rdzv_backend=c10d --rdzv_endpoint=${MASTER_ADDR}:${MASTER_PORT} ${CPM_PATH}/apps/cpm9g/sft_cpm9g_delta.py ${OPTS}"
|
||||
|
||||
echo "${CMD}"
|
||||
$CMD
|
Loading…
Reference in New Issue