#!/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"