[PyTorch] torch.nn.init.xavier_uniform 가중치 초기화 torch 에서 xavier 가중치 초기화 기능을 제공한다. def init_weights(m): if type(m) == nn.Linear: torch.nn.init.xavier_uniform_(m.weight) m.bias.data.fill_(0.01) Python/PyTorch 공부 2021.11.02