trufor-ph2 / scripts /run_trufor_ph3_gpu2.sh
benjaik's picture
Add phase 3: scripts/run_trufor_ph3_gpu2.sh
589f22f verified
Raw
History Blame Contribute Delete
420 Bytes
#!/usr/bin/env bash
set -euo pipefail
PROJECT_ROOT="/data/users/summer2026/trufor/TruFor_train_test"
LOG_DIR="/data/users/summer2026/trufor/training_logs"
LOG_FILE="$LOG_DIR/trufor_ph3_gpu2.log"
mkdir -p "$LOG_DIR"
cd "$PROJECT_ROOT"
export PYTORCH_CUDA_ALLOC_CONF="max_split_size_mb:128"
conda run --no-capture-output -n trufor_cuda102 \
python -u train.py -g 2 -exp trufor_ph3_gpu2 \
2>&1 | tee -a "$LOG_FILE"