#!/bin/bash
cd "$(dirname "$0")"/..
common_command="python -m train ++trainer.max_epochs=600 ++dataset.num_seq=20000 ++model.d_model=32 ++dataset.generator.debug=True"

CUDA_VISIBLE_DEVICES=0 $common_command  experiment=timeseries/ts_id.yaml ++wandb.name=ts_id_T-0 ++model.layer.T=0 &
CUDA_VISIBLE_DEVICES=7 $common_command  experiment=timeseries/ts_ff.yaml ++wandb.name=ts_ff_T-0_l1 ++model.layer.T=0 ++model.layer.n_layers=1&
CUDA_VISIBLE_DEVICES=1 $common_command  experiment=timeseries/ts_lc.yaml ++wandb.name=ts_lc &
CUDA_VISIBLE_DEVICES=2 $common_command  experiment=timeseries/ts_ff.yaml ++wandb.name=ts_ff_T-1 ++model.layer.T=1 &
CUDA_VISIBLE_DEVICES=3 $common_command  experiment=timeseries/ts_id.yaml ++wandb.name=ts_id_T-2 ++model.layer.T=2 &
CUDA_VISIBLE_DEVICES=4 $common_command  experiment=timeseries/ts_ff.yaml ++wandb.name=ts_ff_T-2 ++model.layer.T=2 &
CUDA_VISIBLE_DEVICES=5 $common_command  experiment=timeseries/ts_id.yaml ++wandb.name=ts_id_T-10 ++model.layer.T=10 ++model.layer.dropout=0.2&
CUDA_VISIBLE_DEVICES=6 $common_command  experiment=timeseries/ts_ff.yaml ++wandb.name=ts_ff_T-10 ++model.layer.T=10 ++model.layer.dropout=0.2&