nn.Linear projection shortcut Resnet


I am trying to use resnet architecture for reinforcement learning. Now in the original paper they use a conv layer to scale down the dimension. I want to use a normal nn.linear layer for that but idk how to accomplish it 100%. How should be the weights and bias initialized? Also If someone knows a paper where resnet is used in reinforcement learning would be helpful. Thank you very much!

1 Like