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

2025-12-17 19:20:19 +01:00 · 2019-05-26 15:15:42 +01:00
parent 3b6e413532
commit 2b7d536da4
1 changed files with 1 additions and 0 deletions
--- a/rl_coach/architectures/tensorflow_components/heads/head.py
+++ b/rl_coach/architectures/tensorflow_components/heads/head.py
@@ -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):