Add head regularization costs to tf.losses (#292)

2026-08-27 14:11:55 +02:00 · 2019-05-26 17:15:42 +03:00
parent 3b6e413532
commit 2b7d536da4
1 changed files with 1 additions and 0 deletions
@@ -158,6 +158,7 @@ class Head(object):
        # add regularizations
        for regularization in self.regularizations:
            self.loss.append(regularization)
+            tf.losses.add_loss(regularization)

    @classmethod
    def path(cls):