Thank you very much for providing the code. When I tried to reproduce your work, line 98 of train.py calls model = top.TOP but there is no such method in ConcepPath, which causes a runtime error. Have I misunderstood something?