The detailed Scripts for training three main tasks: predcls, sgcls and sgdet.

[Haeyong Kang]

PySGG/scripts/rel_train_BGNN_vg.sh

Line 9 in c30a6a1

 python -m torch.distributed.launch --master_port 10028 --nproc_per_node=$gpu_num \ 

According to https://github.com/KaihuaTang/Scene-Graph-Benchmark.pytorch,
For Predicate Classification (PredCls), we need to set:
MODEL.ROI_RELATION_HEAD.USE_GT_BOX True MODEL.ROI_RELATION_HEAD.USE_GT_OBJECT_LABEL True

However, occurs an error at

PySGG/pysgg/modeling/roi_heads/relation_head/model_bgnn.py

Line 813 in c30a6a1

 pre_cls_logits, pred_relatedness_scores = self.relation_conf_aware_models[ 

Could you provide the exact scripts for predcls and sgcls modes?

[Rongjie Li]

Hi!

I have added the patch for those two sub-tasks.
The detailed script for running is here. https://github.com/SHTUPLUS/PySGG/blob/main/scripts/rel_train_BGNN_vg_predcls.sh

Thanks for your indication!