Team Ai
Apppublic

MLVLab/Human_Object_Interaction

sourceHugging Faceapache-2.0updated 3y agoView on Hugging Face
1likes
vcoco_train.sh42 linesDownload Raw Back to configs
1#!/usr/bin/env bash2 3set -x4 5EXP_DIR=logs_run_0016PY_ARGS=${@:1}7 8python -u main.py \9    --project_name CPC_HOTR_VCOCO \10    --run_name ${EXP_DIR} \11    --HOIDet \12    --validate \13    --share_enc \14    --pretrained_dec \15    --use_consis \16    --share_dec_param \17    --epochs 90 \18    --lr_drop 60 \19    --lr 1e-4 \20    --lr_backbone 1e-5 \21    --ramp_up_epoch 30 \22    --path_id 0 \23    --num_hoi_queries 16 \24    --set_cost_idx 10 \25    --hoi_idx_loss_coef 1 \26    --hoi_act_loss_coef 10 \27    --backbone resnet50 \28    --hoi_consistency_loss_coef 1 \29    --hoi_idx_consistency_loss_coef 1 \30    --hoi_act_consistency_loss_coef 10 \31    --stop_grad_stage \32    --hoi_eos_coef 0.1 \33    --temperature 0.05 \34    --no_aux_loss \35    --hoi_aux_loss \36    --dataset_file vcoco \37    --data_path v-coco \38    --frozen_weights https://dl.fbaipublicfiles.com/detr/detr-r50-e632da11.pth \39    --output_dir checkpoints/vcoco/ \40    --augpath_name [\'p2\',\'p3\',\'p4\'] \41    ${PY_ARGS}42