Function _ortho

3-atari/2-ppo.py:34–37

(layer, gain)

def _ortho(layer, gain):
    nn.init.orthogonal_(layer.weight, gain)
    nn.init.zeros_(layer.bias)
    return layer


# Nature CNN shared trunk + policy and value heads.