PRIMA-demo / train.sh
HF Space deploy
Deploy snapshot (LFS for demo images per .gitattributes)
c3a4f1d
raw
history blame contribute delete
424 Bytes
#!/bin/bash
set -e
exp_name_stage1=primaStage1
exp_name_stage2=primaStage2
experiment1=primaStage1
experiment2=primaStage2
HYDRA_FULL_ERROR=1 python train.py exp_name=$exp_name_stage1 experiment=$experiment1 trainer=gpu launcher=local
cp -r ./logs/train/runs/$exp_name_stage1 ./logs/train/runs/$exp_name_stage2
HYDRA_FULL_ERROR=1 python train.py exp_name=$exp_name_stage2 experiment=$experiment2 trainer=gpu launcher=local