H2OTest / distributed_train.sh
elineve's picture
Upload 301 files
07423df
raw
history blame contribute delete
80 Bytes
#!/bin/bash
NUM_PROC=$1
shift
torchrun --nproc_per_node=$NUM_PROC train.py "$@"