e07f9a24e60a8acc756efc0728adb257d0474c2f,catalyst/rl/agent/policy.py,CategoricalPolicy,forward,#CategoricalPolicy#Any#Any#Any#,13

Before Change

            not isinstance(logprob, bool) and logprob is not None
        if flag_bool or flag_value:
            // @TODO: refactor
            log_pi = dist.log_prob(logprob)
            return action, log_pi
        return action


class GaussPolicy(nn.Module):

After Change

            if deterministic \
            else dist.sample()

        return _distribution_forward(dist, action, logprob)


class BernoulliPolicy(nn.Module):
    def forward(self, logits, logprob=None, deterministic=False):

Instances