24 lines
774 B
Bash
24 lines
774 B
Bash
#!/usr/bin/env bash
|
|
|
|
set -x
|
|
T=`date +%m%d%H%M`
|
|
|
|
EXP_DIR=exps/singlebaseline/r50_e8_nf4_ld6,7_lr0.0002_nq300_bs4_wbox_joint_MEGA_detrNorm_class31_pretrain_coco_dc5
|
|
mkdir ${EXP_DIR}
|
|
PY_ARGS=${@:1}
|
|
python -u main.py \
|
|
--epochs 8 \
|
|
--num_feature_levels 1\
|
|
--num_queries 300 \
|
|
--dilation \
|
|
--batch_size 4 \
|
|
--num_workers 8 \
|
|
--lr_drop_epochs 6 7 \
|
|
--with_box_refine \
|
|
--dataset_file vid_single \
|
|
--output_dir ${EXP_DIR} \
|
|
--coco_pretrain \
|
|
--resume ./exps/our_models/COCO_pretrained_model/r50_deformable_detr_single_scale_dc5-checkpoint.pth \
|
|
${PY_ARGS} 2>&1 | tee ${EXP_DIR}/log.train.$T
|
|
# --resume /mnt/lustre/helu/code/vod/video_object_detection/exps/pretrainModel/r50_deformable_detr_single_scale_dc5-checkpoint.pth \
|