Visdom shows only the loss. The viewer shows only the predicted cloud…

… now
RaduAlexandru · Jan 23, 2021 · cbedd36 · cbedd36
1 parent bb6ee22
commit cbedd36
Show file tree

Hide file tree

Showing 7 changed files with 5 additions and 5 deletions.
diff --git a/latticenet_py/callbacks/__pycache__/state_callback.cpython-36.pyc b/latticenet_py/callbacks/__pycache__/state_callback.cpython-36.pyc
diff --git a/latticenet_py/callbacks/__pycache__/viewer_callback.cpython-36.pyc b/latticenet_py/callbacks/__pycache__/viewer_callback.cpython-36.pyc
diff --git a/latticenet_py/callbacks/__pycache__/visdom_callback.cpython-36.pyc b/latticenet_py/callbacks/__pycache__/visdom_callback.cpython-36.pyc
diff --git a/latticenet_py/callbacks/viewer_callback.py b/latticenet_py/callbacks/viewer_callback.py
@@ -9,7 +9,7 @@ def __init__(self):
 
     def after_forward_pass(self, pred_softmax, cloud, **kwargs):
         self.show_predicted_cloud(pred_softmax, cloud)
-        self.show_difference_cloud(pred_softmax, cloud)
+        # self.show_difference_cloud(pred_softmax, cloud)
 
 
     def show_predicted_cloud(self, pred_softmax, cloud):

diff --git a/latticenet_py/callbacks/visdom_callback.py b/latticenet_py/callbacks/visdom_callback.py
@@ -10,11 +10,11 @@ def __init__(self, experiment_name):
 
     def after_forward_pass(self, phase, loss, loss_dice, lr, pred_softmax, target, cloud, **kwargs):
         self.vis.log(phase.iter_nr, loss, "loss_"+phase.name,  "loss_"+phase.name+"_"+self.experiment_name, smooth=True,  show_every=10, skip_first=10)
-        self.vis.log(phase.iter_nr, loss_dice, "loss_dice_"+phase.name, "loss_"+phase.name+"_"+self.experiment_name, smooth=True,  show_every=10, skip_first=10)
-        if phase.grad:
-            self.vis.log(phase.iter_nr, lr, "lr", "loss_"+phase.name+"_"+self.experiment_name, smooth=False,  show_every=30)
+        # self.vis.log(phase.iter_nr, loss_dice, "loss_dice_"+phase.name, "loss_"+phase.name+"_"+self.experiment_name, smooth=True,  show_every=10, skip_first=10)
+        # if phase.grad:
+            # self.vis.log(phase.iter_nr, lr, "lr", "loss_"+phase.name+"_"+self.experiment_name, smooth=False,  show_every=30)
 
 
     def epoch_ended(self, phase, **kwargs):
         mean_iou=phase.scores.avg_class_iou(print_per_class_iou=False)
-        self.vis.log(phase.epoch_nr, mean_iou, "iou_"+phase.name,  "loss_"+phase.name+"_"+self.experiment_name, smooth=False,  show_every=1)
+        # self.vis.log(phase.epoch_nr, mean_iou, "iou_"+phase.name,  "loss_"+phase.name+"_"+self.experiment_name, smooth=False,  show_every=1)
diff --git a/latticenet_py/lattice/__pycache__/lattice_funcs.cpython-36.pyc b/latticenet_py/lattice/__pycache__/lattice_funcs.cpython-36.pyc
diff --git a/latticenet_py/lattice/__pycache__/lattice_modules.cpython-36.pyc b/latticenet_py/lattice/__pycache__/lattice_modules.cpython-36.pyc