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
相關文章
相關標籤/搜索