/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
examples/tutorial/new_api/cifar_vit/test_ci.sh
10 строк
254 B
Hongxin Liu
[example] add train resnet/vit with booster example (#3694)
08 май 2023, 05:42
Не верифицирован
08 май 2023, 05:42
f83ea81
Код
Авторство
О чём код?
#!/bin/bash set -xe export DATA=/data/scratch/cifar-10 pip install -r requirements.txt for plugin in "torch_ddp" "torch_ddp_fp16" "low_level_zero"; do colossalai run --nproc_per_node 4 train.py --interval 0 --target_acc 0.83 --plugin $plugin done