Change logging for tensorboard exports (aws#372)

* Change logging for tensorboard stuff

* reword log messagE

Author: rahul003

tornasole/core/hook.py

@@ -117,6 +117,11 @@ def __init__(
 
         self.logger = logger
 
+        if self.tensorboard_dir is None:
+            self.logger.info(
+                f"`tensorboard_dir` has not been set. Disabling export of tensorboard summaries."
+            )
+
         if include_regex is not None:
             collection_manager.get(CollectionKeys.DEFAULT).include(include_regex)
             if CollectionKeys.DEFAULT not in self.include_collections:

tornasole/mxnet/hook.py

@@ -112,8 +112,6 @@ def _export_model(self):
                 tb_writer = self._maybe_get_tb_writer()
                 if tb_writer:
                     tb_writer.write_graph(_net2pb(self.model))
-                else:
-                    self.logger.warning("Model export failed because hook.tensorboard_dir is None")
             except (RuntimeError, TypeError) as e:
                 self.logger.warning(
                     f"Could not export model graph for tensorboard "

tornasole/tensorflow/hook.py

@@ -335,8 +335,6 @@ def _export_model(self):
         tb_writer = self._maybe_get_tb_writer()
         if tb_writer:
             tb_writer.write_graph(self.graph.as_graph_def(add_shapes=True))
-        else:
-            self.logger.debug("Graph not exported because `hook.tensorboard_dir` is None")
         # don't close writer as it might be needed in the step that follows
         # else we will have to open the file again