(labels, logits, weights=None)
| 1842 | |
| 1843 | def _wrap_custom_loss_fn(loss_fn): |
| 1844 | def _wrapper(labels, logits, weights=None): |
| 1845 | if weights is None: |
| 1846 | loss = loss_fn(labels, logits) |
| 1847 | else: |
| 1848 | loss = loss_fn(labels, logits, weights) |
| 1849 | return loss, loss |
| 1850 | return _wrapper |
| 1851 | |
| 1852 |