Register scalar loss so that it will be gathered by :py:`_gather_loss` Args: var_name: loss name (str) or list of loss names
(
self,
var_name: T.Union[str, T.List[str]],
)
| 1177 | self._register(var_name=var_name, target=self.outputs_name) |
| 1178 | |
| 1179 | def _register_loss( |
| 1180 | self, |
| 1181 | var_name: T.Union[str, T.List[str]], |
| 1182 | ): |
| 1183 | """ |
| 1184 | Register scalar loss so that it will be gathered by :py:`_gather_loss` |
| 1185 | |
| 1186 | Args: |
| 1187 | var_name: |
| 1188 | loss name (str) or list of loss names |
| 1189 | """ |
| 1190 | self._register(var_name=var_name, target=self.losses_name) |
| 1191 | |
| 1192 | def _register( |
| 1193 | self, |