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 ·
1258 阅读 ·
0 评论