#!/bin/bash

/root/miniconda3/envs/TEMPO/bin/torchrun \
         --nproc_per_node=6 \
         --nnodes=1 \
         --node_rank=0 \
         --master_addr="localhost" \
         --master_port=29500 \
         ddp.py \
         --train_split /mdcath/train.csv \
         --val_split /mdcath/val.csv \
         --data_dir /data \
         --epochs 5000 \
         --num_frames 20 \
         --threshold 2 \
         --crop 240 \
         --batch_size 6 \
         --val_freq 100 \
         --ckpt_freq 1000 \
         --run_name train_high_resolution \
         --lr 1e-4 \
         --prepend_ipa \
         --time_model rnn \
         --dropout 0.3 \
         --no_offsets \
         --suffix i40 \
         --auxiliary \