def _def_loss(self, model_fn, env):
    '\n        returns a module for and the loss\n        '
    raise NotImplementedError
returns a module for and the loss
def _def_loss(self, model_fn, env):
    '\n        \n        '
    raise NotImplementedError