File size: 424 Bytes
cdad419
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
#!/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