SchNet+attention代码阅读记录
阅读SchNet的过程中,关于一些变量的定义和含义总是忘记,下面做一下记录和整理。class SchNet(torch.nn.Module): def __init__(self, hidden_channels, num_filters, num_interactions, num_gaussians, cutoff, readout='add', dipole, mean, std, atomref): super(SchNet, self).__init__() ... d
原创
2021-09-15 17:23:50 ·
1285 阅读 ·
0 评论