(inputs)
| 157 | """ |
| 158 | |
| 159 | def step_fn(inputs): |
| 160 | losses = self.multi_task.joint_train_step( |
| 161 | inputs, |
| 162 | multi_task_model=self.multi_task_model, |
| 163 | optimizer=self.optimizer, |
| 164 | task_metrics=self.training_metrics) |
| 165 | for key, loss in losses.items(): |
| 166 | self.training_losses[key].update_state(loss) |
| 167 | self.global_step.assign_add(1) |
| 168 | |
| 169 | self.strategy.run( |
| 170 | step_fn, args=(tf.nest.map_structure(next, iterator_map),)) |
nothing calls this directly
no test coverage detected