multi_gpu.sh

#!/bin/bash
#SBATCH --nodes=1
#SBATCH --cpus-per-task=8
#SBATCH --time=5:00:00
#SBATCH --mem=16gb
#SBATCH --partition=visu


cd $(ws_find arnaud)
cd code/flow_pred/

module load devel/cuda/10.1

MASK="" EPOCHS=40 CUDA_VISIBLE_DEVICES=0 ./train_models.sh --unet -c -n 32 -l 5 -p -np -m 5
MASK="" EPOCHS=40 CUDA_VISIBLE_DEVICES=1 ./train_models.sh --unet -c -n 32 -l 5 -p -np -m 5
MASK="" EPOCHS=40 CUDA_VISIBLE_DEVICES=2 ./train_models.sh --unet -c -n 32 -l 5 -p -np -m 5
MASK="" EPOCHS=40 CUDA_VISIBLE_DEVICES=3 ./train_models.sh --unet -c -n 32 -l 5 -p -np -m 5