Closed vieting closed 3 years ago
Add kaiming_normal_ and constant_.
kaiming_normal_
constant_
Also, torch.Tensor.uniform_() and related should have a return value, e.g. I encountered a statement self.mask_emb = nn.Parameter(torch.FloatTensor(args.encoder_embed_dim).uniform_()) which does not work if there is no return value.
torch.Tensor.uniform_()
self.mask_emb = nn.Parameter(torch.FloatTensor(args.encoder_embed_dim).uniform_())
Add
kaiming_normal_
andconstant_
.Also,
torch.Tensor.uniform_()
and related should have a return value, e.g. I encountered a statementself.mask_emb = nn.Parameter(torch.FloatTensor(args.encoder_embed_dim).uniform_())
which does not work if there is no return value.