From 2b7d536da4889337df3223733ed153b1fd496205 Mon Sep 17 00:00:00 2001 From: James Casbon Date: Sun, 26 May 2019 15:15:42 +0100 Subject: [PATCH] Add head regularization costs to tf.losses (#292) --- rl_coach/architectures/tensorflow_components/heads/head.py | 1 + 1 file changed, 1 insertion(+) diff --git a/rl_coach/architectures/tensorflow_components/heads/head.py b/rl_coach/architectures/tensorflow_components/heads/head.py index 397c8ab..d898362 100644 --- 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):