run_train.sh

#!/bin/bash

CUDA_VISIBLE_DEVICES=1,2,3,5,6,7 torchrun --nproc_per_node=6 train_deep_wandb.py --perturbation reverse_full --train_set 10M --batch_size 3 --epoch 7 --seed 0

if [ $? -eq 0 ]; then
  echo "First script completed successfully."
else
  echo "First script failed."
  exit 1
fi

CUDA_VISIBLE_DEVICES=1,2,3,5,6,7 torchrun --nproc_per_node=6 train_deep_wandb.py --perturbation reverse_partial --train_set 10M --batch_size 3 --epoch 7 --seed 0

if [ $? -eq 0 ]; then
  echo "Second script completed successfully."
else
  echo "Second script failed."
  exit 1
fi

CUDA_VISIBLE_DEVICES=1,2,3,5,6,7 torchrun --nproc_per_node=6 train_deep_wandb.py --perturbation reverse_control --train_set 10M --batch_size 3 --epoch 7 --seed 0

if [ $? -eq 0 ]; then
  echo "Third script completed successfully."
else
  echo "Third script failed."
  exit 1
fi