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