Fix layer size in DiagGaussian
Fix layer size in DiagGaussian

I believe this is a bug. Ran into it running the CNN policy on an environment with continuous actions. You have a number of values which are hardcoded (eg: layer sizes) when they should be based on the input/output size.
ikostrikov/pytorch-a2c-ppo-acktr