changed to view loss plotter

AOS55 · Oct 25, 2022 · 7950f4f · 7950f4f
1 parent d3fca9e
commit 7950f4f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/libraries/latentsafesets/utils/loss_plotter.py b/libraries/latentsafesets/utils/loss_plotter.py
@@ -29,7 +29,7 @@ def plot(self):
         for key in self.data:
             losses = self.data[key]
             fname = os.path.join(self.logdir, '%s_loss.pdf' % key)
-            # pu.loss_plot(losses, fname, title='%s loss' % key)
+            pu.loss_plot(losses, fname, title='%s loss' % key)
 
     def print(self, i=None, other_data=None):
         if other_data is None: