Closed wulele2 closed 8 months ago
Hello, i just want to try "mim trick " with ape decoder, my config is : `#!/usr/bin/env bash
set -x
FILE_NAME=$(basename $0) EXP_DIR=./exps/${FILE_NAME%.*} PY_ARGS=${@:1}
python -u main.py \ --output_dir ${EXP_DIR} \ --with_box_refine \ --two_stage \ --mixed_selection \ --look_forward_twice \ --num_queries_one2one 300 \ --num_queries_one2many 1500 \ --k_one2many 6 \ --lambda_one2many 1.0 \ --dropout 0.0 \ --norm_type pre_norm \ --backbone swin_v2_small_window12to16_2global \ --drop_path_rate 0.1 \ --upsample_backbone_output \ --upsample_stride 16 \ --num_feature_levels 1 \ --decoder_type global_ape \ --proposal_feature_levels 4 \ --proposal_in_stride 16 \ --pretrained_backbone_path ./pt_models/swinv2_small_1k_500k_mim_pt.pth \ --epochs 12 \ --lr_drop 11 \ --warmup 1000 \ --lr 2e-4 \ --use_layerwise_decay \ --lr_decay_rate 0.9 \ --weight_decay 0.05 \ --wd_norm_mult 0.0 \ ${PY_ARGS} ` But in the 5 epoch, there is an error: 2024-03-05 04:10 File "/gemini/data-2/Algorithm/Plain-DETR/util/box_ops.py", line 70, in generalized_box_iou 2024-03-05 04:10 assert (boxes1[:, 2:] >= boxes1[:, :2]).all() 2024-03-05 04:10 AssertionError
My env is 8 card 3090; And my command is : bash tools/run_dist_launch.sh 8 configs/swinv2_small_mim_pt_ape.sh Could u give me some advice? Be grateful.
Hello, i just want to try "mim trick " with ape decoder, my config is : `#!/usr/bin/env bash
set -x
FILE_NAME=$(basename $0) EXP_DIR=./exps/${FILE_NAME%.*} PY_ARGS=${@:1}
python -u main.py \ --output_dir ${EXP_DIR} \ --with_box_refine \ --two_stage \ --mixed_selection \ --look_forward_twice \ --num_queries_one2one 300 \ --num_queries_one2many 1500 \ --k_one2many 6 \ --lambda_one2many 1.0 \ --dropout 0.0 \ --norm_type pre_norm \ --backbone swin_v2_small_window12to16_2global \ --drop_path_rate 0.1 \ --upsample_backbone_output \ --upsample_stride 16 \ --num_feature_levels 1 \ --decoder_type global_ape \ --proposal_feature_levels 4 \ --proposal_in_stride 16 \ --pretrained_backbone_path ./pt_models/swinv2_small_1k_500k_mim_pt.pth \ --epochs 12 \ --lr_drop 11 \ --warmup 1000 \ --lr 2e-4 \ --use_layerwise_decay \ --lr_decay_rate 0.9 \ --weight_decay 0.05 \ --wd_norm_mult 0.0 \ ${PY_ARGS} ` But in the 5 epoch, there is an error: 2024-03-05 04:10 File "/gemini/data-2/Algorithm/Plain-DETR/util/box_ops.py", line 70, in generalized_box_iou 2024-03-05 04:10 assert (boxes1[:, 2:] >= boxes1[:, :2]).all() 2024-03-05 04:10 AssertionError
My env is 8 card 3090; And my command is : bash tools/run_dist_launch.sh 8 configs/swinv2_small_mim_pt_ape.sh Could u give me some advice? Be grateful.