#!/bin/bash

CFG=$1
GPUS=$2
AUTORESUME=$3

# add pythonpath
source ./docker/prepare_docker_env.sh

# softlink
ln -sf /lustre/fsw/portfolios/nvr/users/xweng/pretrained ./ckpts
ln -sf /lustre/fsw/portfolios/nvr/projects/nvr_av_end2endav/paradrive/data ./data

# run training
./scripts/e2e_dist_train.sh $CFG $GPUS $AUTORESUME