model VCTREE training too slow

Open rafa-cxg opened this issue 3 years ago • 1 comments

when i train vctree predictor, i find that train 50 iters need 30mins!, which normally been between 2-5min. I do not know why.

Thank you

Mar 02 '22 02:03 rafa-cxg

Hi, For vctree, I simply adopt the implementation from Scene-Graph-Benchmark. It does slow than other model. You can try to comment the torch.autograd.set_detect_anomaly(True) in line 58 of tools/relation_train_net.py. It may slow down backward in some versions of Pytorch. (Refer from this issue https://github.com/SHTUPLUS/PySGG/issues/3 )

Apr 03 '22 06:04 Scarecrow0