-
I am trying to debug my model which is not working properly, and I asked myself a question regarding weight initialization. Usually in other architecture (e.g. CNN model) we initialize the layers weight in the model class. I do not find that in any tutorial about GNNs, so I was wondering where does this take place? Am i supposed to do the initialization somewhere? I suppose it is done automatically but i would like to be sure about it. |
Beta Was this translation helpful? Give feedback.
Replies: 1 comment 2 replies
-
Yes, initialization is done automatically as in regular PyTorch. You can customize them by overriding |
Beta Was this translation helpful? Give feedback.
Yes, initialization is done automatically as in regular PyTorch. You can customize them by overriding
reset_parameters()
or following this thread.