File size: 81 Bytes
786f6a6 |
1 2 3 4 5 6 |
#!/bin/bash
NUM_PROC=$1
shift
torchrun --nproc_per_node=$NUM_PROC train.py "$@"
|
786f6a6 |
1 2 3 4 5 6 |
#!/bin/bash
NUM_PROC=$1
shift
torchrun --nproc_per_node=$NUM_PROC train.py "$@"
|