ColossalAI/examples/vit_b16_imagenet_data_parallel/scripts/train_slurm.sh

3 lines
83 B
Bash
Raw Normal View History

#!/usr/bin/env bash
python train.py --host $HOST --config ./config.py --port 29500