nku-shengzheliu / Pytorch-TransVG

An unofficial pytorch implementation of "TransVG: End-to-End Visual Grounding with Transformers".
MIT License
51 stars 4 forks source link

Issues with test data loader #10

Open preetom-saha-arko opened 1 year ago

preetom-saha-arko commented 1 year ago

How to evaluate the model on test set? When I run the command "python train.py --resume {saved_model_path} --dataset XXX --test", I am getting "Keyerror : 0". This error generates from the getitem function of test data loader.

Moreover, phrase_out and word_split are getting returned from the getitem function but not getting used in test_epoch function (located in train.py). Then what's the point of returning these two?

preetom-saha-arko commented 1 year ago

@nku-shengzheliu Can we have some chat regarding this issue? My email id: arko@cse.buet.ac.bd