You can use the torch.nn.init methods to initialize the parameters as shown in e.g. this post.
1 Like
You can use the torch.nn.init methods to initialize the parameters as shown in e.g. this post.