I am trying to replicate a TF code in PyTorch. I see that in linear layer the initialization is done as:

initialization=‘he’

However I am not able to find a PyTorch equivalent of that. All I get is xavier as:

nn.init.xavier_…(layer.weight)