File size: 847 Bytes
54f7697 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
#!/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 |