File size: 345 Bytes
5806e12 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
#!/bin/bash
source ~/workspace/miniconda3/etc/profile.d/conda.sh
conda activate SaTpy10
export CUDA_VISIBLE_DEVICES=1
export PYTHONPATH="/home/easgrad/shuweiho/workspace/volen/SATE_v1.5/segmentation/models/SaT_cunit_with_maze/wtpsplit:$PYTHONPATH"
CONFIG_FILE="lora_sat_12l.json"
python wtpsplit/wtpsplit/train/train_lora.py "$CONFIG_FILE"
|