RMeli / gnina-torch

🔥 PyTorch implementation of GNINA scoring function for molecular docking

Home Page:https://gnina-torch.rtfd.io

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Common weights initialisation

RMeli opened this issue · comments

Define weights and biases initialisation function outside of models' __init__ and perform initialisation with nn.Module.apply() function.

Check how biases are initialised and perform the same initialisation.