Shuwei Hou
initial_for_hf
5806e12
raw
history blame contribute delete
698 Bytes
#!/bin/bash
# SaT 3-layer Full Parameter Fine-tuning Script
# Author: Generated for ENNI dataset fine-tuning
echo "Starting SaT 3-layer full parameter fine-tuning..."
source ~/workspace/miniconda3/etc/profile.d/conda.sh
conda activate SaTpy10
# Set CUDA device (adjust if needed)
export CUDA_VISIBLE_DEVICES=0
export PYTHONPATH="/home/easgrad/shuweiho/workspace/volen/SATE_v1.5/segmentation/models/SaT_cunit_with_maze/wtpsplit:$PYTHONPATH"
# Run training with full parameter fine-tuning using our custom script
python wtpsplit/wtpsplit/train/train_full_param.py sat_3l_full_finetune.json
echo "SaT 3-layer full fine-tuning completed!"
echo "Model saved to: ./model_finetuned/sat-3l_full_ENNI"