コード例 #1
0
ファイル: program.py プロジェクト: yyqgood/tensorboard
    def _register_info(self, server):
        """Write a TensorBoardInfo file and arrange for its cleanup.

        Args:
          server: The result of `self._make_server()`.
        """
        server_url = urllib.parse.urlparse(server.get_url())
        info = manager.TensorBoardInfo(
            version=version.VERSION,
            start_time=int(time.time()),
            port=server_url.port,
            pid=os.getpid(),
            path_prefix=self.flags.path_prefix,
            logdir=self.flags.logdir or self.flags.logdir_spec,
            db=self.flags.db,
            cache_key=self.cache_key,
        )
        atexit.register(manager.remove_info_file)
        manager.write_info_file(info)
コード例 #2
0
def _make_info(i=0):
    """Make a sample TensorBoardInfo object.

  Args:
    i: Seed; vary this value to produce slightly different outputs.

  Returns:
    A type-correct `TensorBoardInfo` object.
  """
    return manager.TensorBoardInfo(
        version=version.VERSION,
        start_time=1548973541 + i,
        port=6060 + i,
        pid=76540 + i,
        path_prefix="/foo",
        logdir="~/my_data/",
        db="",
        cache_key="asdf",
    )