hiyouga 52cc6bce38 fix #3602
Former-commit-id: b0888262e371280f2b69ab529040b3a99632fc12
2024-05-07 17:50:27 +08:00

10 lines
231 B
Bash

#!/bin/bash
NPROC_PER_NODE=4
CUDA_VISIBLE_DEVICES=0,1,2,3 python -m torch.distributed.run \
--nproc_per_node $NPROC_PER_NODE \
--nnodes 1 \
--standalone \
src/train.py examples/full_multi_gpu/llama3_full_sft.yaml