File size: 377 Bytes
0b5ed64 | 1 2 3 4 5 6 7 8 | #!/bin/bash
gpu_id=$1
task_name=$2
experiment=$3
tmux new-session -d -s "train_${gpu_id}_${task_name}_${experiment}"
tmux send-keys -t "train_${gpu_id}_${task_name}_${experiment}" "conda activate pfp_env && WANDB__SERVICE_WAIT=300 CUDA_VISIBLE_DEVICES=$gpu_id python scripts/train.py log_wandb=True dataloader.num_workers=8 task_name=$task_name +experiment=$experiment" Enter
|