SnakeCLEF2024 / pytorch-image-models /distributed_train.sh
pengdadaaa's picture
Upload 741 files
786f6a6 verified
raw
history blame contribute delete
81 Bytes
#!/bin/bash
NUM_PROC=$1
shift
torchrun --nproc_per_node=$NUM_PROC train.py "$@"