AssertionError: Non-existent key: MODEL.ROI_RELATION_HEAD.MP_LAYER_NUM #3
Description
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!