16*16

[erjpc]

Hello author, when I tried to extract 16*16 features
python extract_trar_grid_feature.py
--config-file configs/X-152-grid.yaml
--dataset coco_2014_train
--output_dir /data_home/jiangpeichao/data/vqa_Grid_feature16/feats
--weight_path /data_home/jiangpeichao/X-152.pth
--feature_size 16
An error occurred：
Traceback (most recent call last):
File "extract_trar_grid_feature.py", line 120, in
main(args)
File "extract_trar_grid_feature.py", line 105, in main
model = build_model(cfg)
File "/data_home/jiangpeichao/detectron2/detectron2/modeling/meta_arch/build.py", line 22, in build_model
model = META_ARCH_REGISTRY.get(meta_arch)(cfg)
File "/data_home/jiangpeichao/detectron2/detectron2/config/config.py", line 189, in wrapped
explicit_args = _get_args_from_config(from_config_func, *args, **kwargs)
File "/data_home/jiangpeichao/detectron2/detectron2/config/config.py", line 245, in _get_args_from_config
ret = from_config_func(*args, **kwargs)
File "/data_home/jiangpeichao/detectron2/detectron2/modeling/meta_arch/rcnn.py", line 77, in from_config
"roi_heads": build_roi_heads(cfg, backbone.output_shape()),
File "/data_home/jiangpeichao/detectron2/detectron2/modeling/roi_heads/roi_heads.py", line 43, in build_roi_heads
return ROI_HEADS_REGISTRY.get(name)(cfg, input_shape)
File "/data_home/jiangpeichao/TRAR-Feature-Extraction-master/grid_feats/roi_heads.py", line 175, in init
super(StandardROIHeads, self).init(cfg, input_shape)
File "/data_home/jiangpeichao/detectron2/detectron2/config/config.py", line 190, in wrapped
init_func(self, **explicit_args)
TypeError: init() got an unexpected keyword argument 'train_on_pred_boxes'
Can you help me with it