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.