diff --git a/modules/deeplearning/unet.py b/modules/deeplearning/unet.py
index 5c66a3cd64cba08afdbad09cc93bac434e113243..39ac89b8fa63b057b12cf21fafa3b77db0063af8 100644
--- a/modules/deeplearning/unet.py
+++ b/modules/deeplearning/unet.py
@@ -548,8 +548,8 @@ class UNET:
         self.initial_learning_rate = initial_learning_rate
 
     def build_evaluation(self):
-        #self.train_loss = tf.keras.metrics.Mean(name='train_loss')
-        #self.test_loss = tf.keras.metrics.Mean(name='test_loss')
+        # self.train_loss = tf.keras.metrics.Mean(name='train_loss')
+        # self.test_loss = tf.keras.metrics.Mean(name='test_loss')
         self.train_accuracy = tf.keras.metrics.MeanAbsoluteError(name='train_accuracy')
         self.test_accuracy = tf.keras.metrics.MeanAbsoluteError(name='test_accuracy')
         self.train_loss = tf.keras.metrics.Mean(name='train_loss')