Manufacturing-MoE / train.sh
rasitcesur's picture
Sync from rasitcesur/Manufacturing-MoE via hub-sync
547ac30 verified
Raw History Blame Contribute Delete
104 Bytes
#!/usr/bin/env bash
set -euo pipefail
cd "$(dirname "$0")"
python train.py --device auto --batch 4 "$@"