Update execute_func.py

revise logging naming
facebookresearch · Nov 12, 2023 · 45e7e3f · 45e7e3f
1 parent 26ac783
commit 45e7e3f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/execute_func.py b/execute_func.py
@@ -749,7 +749,7 @@ def log(self, mode, inputs, outputs, losses):
 
                 writer.add_image(
                     "out_{}/{}".format(s, j),
-                    normalize_image(outputs[("out", s)][j]), self.step)
+                    normalize_image(outputs[("inverse_depth_output", s)][j]), self.step)
 
                 # automasking
                 writer.add_image(
@@ -1100,4 +1100,4 @@ def load_model_multi(self):
             except ValueError:
                 print("Can't load Adam - using random")
         else:
-            print("Cannot find Adam weights so Adam is randomly initialized")
+            print("Cannot find Adam weights so Adam is randomly initialized")