search

yunshengb / SimGNN

146 stars 33 forks source link

Error when running main.py under IMDBMulti dataset #8

Open SaizhuoWang opened 5 years ago

SaizhuoWang commented 5 years ago

Hi! Thank you very much for sharing the code. Sorry to disturb, but I have met some problems when running main.py under IMDBMulti dataset. The program works fine under LINUX and AIDS700nef datasets. But it has some errors under IMDBMulti. My command is:

CUDA_VISIBLE_DEVICES=1 python main.py --dataset imdbmulti --gpu 1

There're 4 GTX 1080Ti GPUs on my server so I am sure the GPU I assigned is valid. And after building the model, when the optimization started, the program errs. And the error messages are like this: image image image image image Sorry that I'm not quite familiar with tensorflow. Could you please tell me what the problem is? Thank you very much!

yunshengb commented 5 years ago

I see some error related to padding. Can you paste the entire configuration here? I suspect it is because of the max_nodes flag set wrongly. If you use config.py, we set it to 90 https://github.com/yunshengb/SimGNN/blob/master/model/Siamese/config.py#L20. If you manually provide command line arguments, make sure to add something like --max_nodes 90.

yunshengb commented 5 years ago

The idea of creating config.py is to avoid typing in command line arguments. Instead, we can modify a few parameters in config.py and let the file handle the setting of the rest parameters via a bunch of if else statements. Then we can simply run python main.py without the following --xxxs. We apologize we did not make this idea clear. We have update the README accordingly. Thank you very much!