run_job.sh

#!/bin/bash

# --- 0. Activate Conda Environment ---
echo "Activating Conda environment..."
source /home/n5huang/miniconda3/etc/profile.d/conda.sh
conda activate dnabert_v2

# --- 1. Configuration ---
# Set the GPU ID
export CUDA_VISIBLE_DEVICES=6
echo "Assigned GPU: $CUDA_VISIBLE_DEVICES"

# --- 2. Correct Working Directory ---
# Move to the folder where the script actually lives to ensure relative paths work (if any)
cd /home/n5huang/dna_token/SparseAE/

# --- 3. Run the Python Script ---
# We use absolute path just to be safe
# -u ensures logs are written immediately
echo "Starting Python training..."
python -u extract.py #run_train.py