#!/bin/bash
cd "$(dirname "$0")"/..
common_command="python -m train experiment=timeseries/ts_id.yaml ++model.layer.T=0 ++trainer.max_epochs=600 ++dataset.generator.debug=True ++dataset.generator.hard=False ++dataset.num_seq=1000 ++dataset.generator.h=-1 ++dataset.generator.k_bias=0.5 ++dataset.generator.a_bias=0.15 ++model.d_model=128"

CUDA_VISIBLE_DEVICES=0 $common_command  experiment=timeseries/ts_id.yaml ++wandb.name=ts_id_T-0_kb-0.5_ab-0.15 ++model.layer.T=0 &
CUDA_VISIBLE_DEVICES=1 $common_command  experiment=timeseries/ts_lc.yaml ++wandb.name=ts_lc_T-0_kb-0.5_ab-0.15  &
CUDA_VISIBLE_DEVICES=2 $common_command  experiment=timeseries/ts_id.yaml ++wandb.name=ts_id_T-1_kb-0.5_ab-0.15 ++model.layer.T=1 &
CUDA_VISIBLE_DEVICES=3 $common_command  experiment=timeseries/ts_id.yaml ++wandb.name=ts_id_T-2_kb-0.5_ab-0.15 ++model.layer.T=2 &
CUDA_VISIBLE_DEVICES=4 $common_command  experiment=timeseries/ts_mha.yaml ++wandb.name=ts_mha_T-0_kb-0.5_ab-0.15 &
CUDA_VISIBLE_DEVICES=5 $common_command  experiment=timeseries/ts_id.yaml ++wandb.name=ts_id_T-10_kb-0.5_ab-0.15 ++model.layer.T=10 &
CUDA_VISIBLE_DEVICES=6 $common_command  experiment=timeseries/ts_id.yaml ++wandb.name=ts_id_T-20_kb-0.5_ab-0.15 ++model.layer.T=20 &

