KaihuaTang / Scene-Graph-Benchmark.pytorch

A new codebase for popular Scene Graph Generation methods (2020). Visualization & Scene Graph Extraction on custom images/datasets are provided. It's also a PyTorch implementation of paper “Unbiased Scene Graph Generation from Biased Training CVPR 2020”
MIT License
1.03k stars 228 forks source link

SGDet model can't train #210

Open Lxy811 opened 2 months ago

Lxy811 commented 2 months ago

Hello, when I was conducting SGDet training, I was either stuck in the training loss or when the verification results were loaded in the model verification. Could you please see that these are two kinds of errors? Is there any suggestion to solve these problems? Two Gpus 3090 are always full of usage. Snipaste_2024-04-13_17-16-52 Snipaste_2024-04-13_17-17-22 Snipaste_2024-04-13_17-17-35