Method
build_losses
(self, labels, model_outputs, aux_losses=None)
Source from the content-addressed store, hash-verified
| 105 | return MockFooModel(shared_layer=tf_keras.layers.Dense(1)) |
| 106 | |
| 107 | def build_losses(self, labels, model_outputs, aux_losses=None) -> tf.Tensor: |
| 108 | loss = tf_keras.losses.mean_squared_error(labels, model_outputs) |
| 109 | if aux_losses: |
| 110 | loss += tf.add_n(aux_losses) |
| 111 | return tf.reduce_mean(loss) |
| 112 | |
| 113 | |
| 114 | @task_factory.register_task_cls(BarConfig) |
Tested by
no test coverage detected