def _export_model(self):
    if (self.model is not None):
        try:
            tb_writer = self._maybe_get_tb_writer()
            if tb_writer:
                tb_writer.write_graph(_net2pb(self.model))
        except (RuntimeError, TypeError) as e:
            self.logger.warning(f'Could not export model graph for tensorboard due to the mxnet exception: {e}')
def _export_model(self):
    if (self.model is not None):
        try:
            tb_writer = self._maybe_get_tb_writer()
            if tb_writer:
                tb_writer.write_graph(_net2pb(self.model))
        except (RuntimeError, TypeError) as e:
            self.logger.warning(f'Could not export model graph for tensorboard due to the mxnet exception: {e}')