R2SE_model / scripts /server /run_train_paradrive.sh

Upload folder using huggingface_hub

663494c verified 3 months ago

323 Bytes

	#!/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