pygcn icon indicating copy to clipboard operation
pygcn copied to clipboard

specifying modes train, validation and test

Open fansariadeh opened this issue 4 years ago • 1 comments

Hi guys!

I hope any of you happen to know why in line 88 of train.py, while it tries to specifies test mode, we see these line: def test(): model.eval()
Dont you think it should be ### model.test() ### sine we are focusing on test set?

Thank you in advance guys.

fansariadeh avatar Apr 19 '20 23:04 fansariadeh

@fansariadeh There's no model.test() in pytorch. When you train the model, call model.train() will activate some layers like batch-norm, zero-out. When you test the model, call model.eval() will modify the operations through these layers.

chao1224 avatar May 07 '20 20:05 chao1224