[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)