def on_start(self, task) -> None:
"""
Plot the model on Tensorboard.
"""
if is_primary():
try:
# Show model in tensorboard:
logging.info("Showing model graph in TensorBoard...")
plot_model(
task.base_model,
size=task.base_model.input_shape,
input_key=task.base_model.input_key if hasattr(
task.base_model, "input_key") else None,
writer=self.tb_writer,
)
except Exception:
logging.warn("Unable to plot model to tensorboard")
logging.debug("Exception encountered:", exc_info=True)
def on_start(self, task: "tasks.ClassyTask") -> None:
"""
Plot the model on Tensorboard.
"""
if is_master():
try:
# Show model in tensorboard:
logging.info("Showing model graph in TensorBoard...")
plot_model(
task.base_model,
size=task.base_model.input_shape,
input_key=task.base_model.input_key if hasattr(
task.base_model, "input_key") else None,
writer=self.tb_writer,
)
except Exception:
logging.warn(
"Unable to plot model to tensorboard. Exception: ",
exc_info=True)
