Spaces:

Sven33
/

SATEv1.5

Starting on L4

App Files Files Community

SATEv1.5 / segmentation /models /SaT_cunit_with_maze /lora_12l.sh

Shuwei Hou

initial_for_hf

5806e12 8 days ago

history blame contribute delete

345 Bytes

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