Voltline's picture
Release VimeML V4.2 epoch20 FP32 and Core ML INT8 (GPL-2.0)
c83c617 verified
Raw History Blame Contribute Delete
398 Bytes
#!/usr/bin/env bash
set -euo pipefail
cd "$(dirname "$0")/../.."
source /root/autodl-tmp/vimeml-wandb.env
export OMP_NUM_THREADS=4
export MKL_NUM_THREADS=4
export PYTHONUNBUFFERED=1
export WANDB_DIR="$PWD/artifacts/tracking"
mkdir -p "$WANDB_DIR"
python_bin="${VIMEML_PYTHON:-/root/miniconda3/bin/python}"
exec "$python_bin" -X utf8 -u scripts/training/train.py --config configs/train-v2.toml "$@"