551436a8106f3292db2f7d1a6da3462f2aa8bb6b,maml_rl/policies/normal_mlp.py,NormalMLPPolicy,forward,#NormalMLPPolicy#Any#Any#,25

Before Change

        sigma = F.linear(output, weight=params["sigma.weight"],
            bias=params["sigma.bias"])

        return Normal(loc=mu, scale=F.softplus(sigma))

After Change

            output = self.nonlinearity(output)
        mu = F.linear(output, weight=params["mu.weight"],
            bias=params["mu.bias"])
        scale = torch.exp(torch.clamp(params["sigma"], min=self.min_log_std))

        return Normal(loc=mu, scale=scale)

Instances