Did you include the code for the self-critics RL training for VCTree?

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.06k stars 228 forks source link

Did you include the code for the self-critics RL training for VCTree? #88

Closed alibabadoufu closed 4 years ago

alibabadoufu commented 4 years ago

❓ Questions and Help

Hi, I am trying to find the code where you train the VCTree with self-critics RL algo, but I could not find it. Could you please kindly advise this?

KaihuaTang commented 4 years ago

I remove the self-critics RL finetuning in this project for simplicity, because I only need to use the VCTree structure as a baseline to test our TDE method.