def _def_loss(self, model_fn):

    class Ent(Module):

        def _build(self, inputs):
            return self.modules[0]._entropy
    return Ent('l', model_fn(self.env))
def _def_loss(self, model_fn):

    class Ent(Module):

        def _build(self, inputs):
            return self.modules[0]._entropy
    return Ent('l', model_fn(self.env))