Defining weight manually for LSTM - PyTorch Forums
discuss.pytorch.org › t › defining-weight-manuallyNov 11, 2020 · Recently I was diving into meta-learning, and need to change the weights of module during the training process, so I can’t use off-the-shelf torch.nn.Conv2d or torch.nn.LSTM module for I can’t pass weights into the module. Instead, I have to define weights manually and call the underlying interface. For convolution layers or batch normalization layers, PyTorch provides torch.nn.Functional ...