File size: 703 Bytes
5806e12 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
#!/bin/bash
# SaT 12-layer Full Parameter Fine-tuning Script
# Author: Generated for ENNI dataset fine-tuning
echo "Starting SaT 12-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_12l_full_finetune.json
echo "SaT 12-layer full fine-tuning completed!"
echo "Model saved to: ./model_finetuned/sat-12l_full_ENNI" |