For my current neural network I’d like to initialize my model’s weights with the torch.nn.init.kaiming_normal, so I have tried this :
torch.nn.init.kaiming_normal(m.weight.data)
but I receive this error message :
Fan in and fan out can not be computed for tensor with less than 2 dimensions
Is the less than a strictly less than ? I am not sure I understand why it’s failing, if someone could give me some pointers that would be great.