def METHOD_NAME(self):
    relu_gain = nn.init.calculate_gain('relu')
    self.conv1.weight.data.mul_(relu_gain)
    self.conv2.weight.data.mul_(relu_gain)
    self.linear0.weight.data.mul_(relu_gain)
    self.linear1.weight.data.mul_(relu_gain)
    self.actor.weight.data.mul_(relu_gain)
    self.critic.weight.data.mul_(relu_gain)