CUDA_DEVICES="0"
MASTER_PORT=9980
NPROC_GPU=1
PRETRAIN_MODEL="../model_zoo/ovtr_5_frame.pth"
OUTPUT="./results"
VIS_OUTPUT="./results"
CUDA_VISIBLE_DEVICES="${CUDA_DEVICES}" python -m torch.distributed.launch --master_port=${MASTER_PORT} --nproc_per_node=${NPROC_GPU} \
    --use_env \
    ./track_demo.py \
    --config_file ./config/ovtr_5_frame_train_val.py \
    --dataset_file lvis_generated_img_seqs \
    --epochs 16 \
    --with_box_refine \
    --two_stage \
    --lr 4e-5 \
    --lr_backbone 4e-6 \
    --lr_drop 13 \
    --pretrain ${PRETRAIN_MODEL} \
    --output_dir ${OUTPUT} \
    --num_workers 32 \
    --batch_size 1 \
    --sample_mode random_interval \
    --sample_interval 1 \
    --sampler_steps 4 7 14 \
    --sampler_lengths 2 3 4 5 \
    --merger_dropout 0 \
    --random_drop 0.1 \
    --fp_ratio 0.3 \
    --track_query_iteration CIP \
    --calculate_negative_samples \
    --score_thresh 0.3 \
    --filter_score_thresh 0.3 \
    --ious_thresh 0.3 \
    --miss_tolerance 5 \
    --maximum_quantity 50 \
    --vis_output ${VIS_OUTPUT}