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"