config_file for train_sg_net.py

microsoft / scene_graph_benchmark

image scene graph generation benchmark

MIT License

390 stars 86 forks source link

config_file for train_sg_net.py #33

Open aaronhd opened 3 years ago

aaronhd commented 3 years ago

Hi,

I try to train a model following "python tools/train_sg_net.py --config-file "/path/to/config/file.yaml" ". But I confuse which config.file I need to use, can you share more details about training config files?

Thank you very much!

hanxiaotian commented 3 years ago

We have provide several config files for different models, like Reldn, IMP, GRCNN, etc. For example, if you want to train a Reldn model, you can use following command: python tools/train_sg_net.py --config-file sgg_configs/vg_vrd/rel_danfeiX_FPN50_reldn.yam