def _sigmoid_output(self, output):
if "hm" in output:
output["hm"] = _sigmoid(output["hm"])
if "hm_hp" in output:
output["hm_hp"] = _sigmoid(output["hm_hp"])
if "dep" in output:
output["dep"] = 1.0 / (output["dep"].sigmoid() + 1e-6) - 1.0
return output
def _sigmoid_output(self, output):
if 'hm' in output:
output['hm'] = _sigmoid(output['hm'])
if 'hm_hp' in output:
output['hm_hp'] = _sigmoid(output['hm_hp'])
if 'dep' in output:
output['dep'] = 1. / (output['dep'].sigmoid() + 1e-6) - 1.
return output
def _sigmoid_output(self, output):
if 'hm' in output:
output['hm'] = _sigmoid(output['hm'])
if 'hm_hp' in output:
output['hm_hp'] = _sigmoid(output['hm_hp'])
if 'dep' in output:
output['dep'] = 1. / (output['dep'].sigmoid() + 1e-6) - 1.
if 'dep_sec' in output and self.opt.sigmoid_dep_sec:
output['dep_sec'] = 1. / (output['dep_sec'].sigmoid() + 1e-6) - 1.
return output