search

siml3 / RU-Net

MIT License
11 stars 3 forks source link

AssertionError: Non-existent key: MODEL.ROI_RELATION_HEAD.MP_LAYER_NUM #3

Open Git-oNmE opened 1 year ago

Git-oNmE commented 1 year ago

Hello, thanks for the solid work. However, when I want to reproduce the SGDet result in below command: PYTHONPATH=$PWD CUDA_VISIBLE_DEVICES=0,1,2,3 python -m torch.distributed.launch --master_port 10022 --nproc_per_node=4 tools/relation_train_net.py --config-file "configs/e2e_relation_X_101_32_8_FPN_1x.yaml" MODEL.ROI_RELATION_HEAD.USE_GT_BOX False MODEL.ROI_RELATION_HEAD.USE_GT_OBJECT_LABEL False MODEL.PRETRAINED_DETECTOR_CKPT checkpoints/pretrained_faster_rcnn/model_final.pth OUTPUT_DIR "exp/RU-Net" SOLVER.IMS_PER_BATCH 12 TEST.IMS_PER_BATCH 4 DTYPE "float16" SOLVER.PRE_VAL True MODEL.ROI_RELATION_HEAD.PREDICTOR RUNetPredictor MODEL.ROI_RELATION_HEAD.MP_LAYER_NUM 5 SOLVER.BASE_LR 0.0025 GLOVE_DIR glove MODEL.MODEL.ROI_RELATION_HEAD.L21_LOSS 0.7 I got error message AssertionError: Non-existent key: MODEL.ROI_RELATION_HEAD.MP_LAYER_NUM.

Do you know how to solve this problem? Thanks!

zbw4034 commented 1 year ago

got same problem. Anyone have some idea?

ZHUXUHAN commented 1 year ago

got same problem. Anyone have some idea?

I was unable to reproduce Predcls and SGcls, too.