gryf/coach
1
0 Fork 0
mirror of https://github.com/gryf/coach.git synced 2026-10-13 18:22:29 +02:00
Code Issues Projects Releases Wiki Activity

summing head losses instead of taking the mean (#98)

Browse Source
This commit is contained in:
Gal Leibovich authored and Gal Novik committed 2018-11-18 12:20:00 +02:00
1 parent 5b11fa5656
commit 449bcfb4e1
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
rl_coach/architectures/tensorflow_components/general_network.py
+1 -1
View File
@@ -280,7 +280,7 @@ class GeneralTensorFlowNetwork(TensorFlowArchitecture):
# Losses # Losses
self.losses = tf.losses.get_losses(self.full_name) self.losses = tf.losses.get_losses(self.full_name)
self.losses += tf.losses.get_regularization_losses(self.full_name) self.losses += tf.losses.get_regularization_losses(self.full_name)
self.total_loss = tf.losses.compute_weighted_loss(self.losses, scope=self.full_name) self.total_loss = tf.reduce_sum(self.losses)
# tf.summary.scalar('total_loss', self.total_loss) # tf.summary.scalar('total_loss', self.total_loss)
# Learning rate # Learning rate