File size: 420 Bytes
589f22f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | #!/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"
|