lsnu's picture
Add files using upload-large-folder tool
0b5ed64 verified
#!/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