Pytorch 参数初始化以及Xavier初始化

def _initialize_weights(self): # print(self.modules()) for m in self.modules(): print(m) if isinstance(m, nn.Linear): # print(m.weight.data.typ
相关文章
相关标签/搜索