File size: 343 Bytes
5806e12
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
#!/bin/bash
source ~/workspace/miniconda3/etc/profile.d/conda.sh

conda activate SaTpy10

export CUDA_VISIBLE_DEVICES=0
export PYTHONPATH="/home/easgrad/shuweiho/workspace/volen/SATE_v1.5/segmentation/models/SaT_cunit_with_maze/wtpsplit:$PYTHONPATH"


CONFIG_FILE="lora_sat_3l.json"

python wtpsplit/wtpsplit/train/train_lora.py "$CONFIG_FILE"