About the training phases

yuweihao / KERN

Code for Knowledge-Embedded Routing Network for Scene Graph Generation (CVPR 2019)

MIT License

120 stars 35 forks source link

Closed kellyiss closed 5 years ago

kellyiss commented 5 years ago

In Step 5 Train scene graph classification: run CUDA_VISIBLE_DEVICES=YOUR_GPU_NUM ./scripts/train_kern_predcls.sh.

a mistake? run ./scripts/train_kern_sgcls.sh ?

yuweihao commented 5 years ago

Hi @kellyiss , thanks for your reminder. It is a typo and I have fixed it. As you guess, it should be run ./scripts/train_kern_sgcls.sh.