File size: 647 Bytes
b46126b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | #!/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
|