11 lines
132 B
Bash
11 lines
132 B
Bash
#!/bin/bash
|
|
|
|
set -a
|
|
|
|
: ${NUM_GPUS:=8}
|
|
: ${GPU_BATCH_SIZE:=72}
|
|
: ${GRAD_ACCUMULATION:=2}
|
|
: ${AMP=:false}
|
|
|
|
bash scripts/train.sh "$@"
|