bdd_source_and_dets18k.sh

#!/usr/bin/env bash
#SBATCH --job-name=bdd_source_and_dets18k
#SBATCH -o gypsum/logs/%j_bdd_source_and_dets18k.txt 
#SBATCH -e gypsum/errs/%j_bdd_source_and_dets18k.txt
#SBATCH -p 1080ti-long
#SBATCH --gres=gpu:1
#SBATCH --mem=100000
##SBATCH --cpus-per-task=4
##SBATCH --mem-per-cpu=4096



python tools/train_net_step.py \
    --dataset bdd_peds+DETS18k \
    --cfg configs/baselines/bdd_peds_dets_bs64_4gpu.yaml  \
    --set NUM_GPUS 1 TRAIN.SNAPSHOT_ITERS 5000 \
    --iter_size 2 \
    --use_tfboard \
    --load_ckpt /mnt/nfs/scratch1/pchakrabarty/bdd_recs/ped_models/bdd_peds.pth \