rlqja1107 / torch-ST-SGG

Official PyTorch implementation Source code for Adaptive Self-Training Framework for Fine-grained Scene Graph generation (ST-SGG), accepted at ICLR 2024
18 stars 0 forks source link

KeyError: 'boxes_per_cls' #2

Open suexin27 opened 5 months ago

suexin27 commented 5 months ago

hi, when I execute the command "bash run_shell/bgnn/predcls/bilvl/pretrain.sh", I met the bug below. What is "boxes_per_cls"? I want to add this key value during preprocessing, but I don't know which variable to assign to it?

image image