Python is_root_procの例

コード例 #1

    def dump_tabular(self):
        """
        Write all of the diagnostics from the current iteration
        """
        if (MPIUtil.is_root_proc()):
            if (self.first_row):
                self._dump_str_template = self._build_str_template()

            vals = []
            for key in self.log_headers:
                val = self.log_current_row.get(key, "")
                vals.append(val)

            if self.output_file is not None:
                if self.first_row:
                    header_str = self._dump_str_template.format(
                        *self.log_headers)
                    self.output_file.write(header_str + "\n")

                val_str = self._dump_str_template.format(*map(str, vals))
                self.output_file.write(val_str + "\n")
                self.output_file.flush()

        self.log_current_row.clear()
        self.first_row = False
        return

コード例 #2

ファイル: logger.py プロジェクト: bulletphysics/bullet3

  def dump_tabular(self):
    """
        Write all of the diagnostics from the current iteration
        """
    if (MPIUtil.is_root_proc()):
      if (self.first_row):
        self._dump_str_template = self._build_str_template()

      vals = []
      for key in self.log_headers:
        val = self.log_current_row.get(key, "")
        vals.append(val)

      if self.output_file is not None:
        if self.first_row:
          header_str = self._dump_str_template.format(*self.log_headers)
          self.output_file.write(header_str + "\n")

        val_str = self._dump_str_template.format(*map(str, vals))
        self.output_file.write(val_str + "\n")
        self.output_file.flush()

    self.log_current_row.clear()
    self.first_row = False
    return

コード例 #3

ファイル: normalizer.py プロジェクト: bulletphysics/bullet3

 def check_synced(self):
   synced = True
   if MPIUtil.is_root_proc():
     vars = np.concatenate([self.mean, self.mean_sq])
     MPIUtil.bcast(vars)
   else:
     vars_local = np.concatenate([self.mean, self.mean_sq])
     vars_root = np.empty_like(vars_local)
     MPIUtil.bcast(vars_root)
     synced = (vars_local == vars_root).all()
   return synced

コード例 #4

ファイル: normalizer.py プロジェクト: takazerker/bullet4unity

 def check_synced(self):
     synced = True
     if MPIUtil.is_root_proc():
         vars = np.concatenate([self.mean, self.mean_sq])
         MPIUtil.bcast(vars)
     else:
         vars_local = np.concatenate([self.mean, self.mean_sq])
         vars_root = np.empty_like(vars_local)
         MPIUtil.bcast(vars_root)
         synced = (vars_local == vars_root).all()
     return synced

コード例 #5

    def configure_output_file(self, filename=None):
        """
        Set output directory to d, or to /tmp/somerandomnumber if d is None
        """
        self.first_row = True
        self.log_headers = []
        self.log_current_row = {}

        output_path = filename or "output/log_%i.txt" % int(time.time())

        out_dir = os.path.dirname(output_path)
        if not os.path.exists(out_dir) and MPIUtil.is_root_proc():
            os.makedirs(out_dir)

        if (MPIUtil.is_root_proc()):
            self.output_file = open(output_path, 'w')
            assert osp.exists(output_path)
            atexit.register(self.output_file.close)

            Logger.print2("Logging data to " + self.output_file.name)
        return

コード例 #6

ファイル: logger.py プロジェクト: bulletphysics/bullet3

  def configure_output_file(self, filename=None):
    """
        Set output directory to d, or to /tmp/somerandomnumber if d is None
        """
    self.first_row = True
    self.log_headers = []
    self.log_current_row = {}

    output_path = filename or "output/log_%i.txt" % int(time.time())

    out_dir = os.path.dirname(output_path)
    if not os.path.exists(out_dir) and MPIUtil.is_root_proc():
      os.makedirs(out_dir)

    if (MPIUtil.is_root_proc()):
      self.output_file = open(output_path, 'w')
      assert osp.exists(output_path)
      atexit.register(self.output_file.close)

      Logger.print2("Logging data to " + self.output_file.name)
    return

コード例 #7

 def print_tabular(self):
     """
     Print all of the diagnostics from the current iteration
     """
     if (MPIUtil.is_root_proc()):
         vals = []
         Logger.print2("-" * 37)
         for key in self.log_headers:
             val = self.log_current_row.get(key, "")
             if isinstance(val, float):
                 valstr = "%8.3g" % val
             elif isinstance(val, int):
                 valstr = str(val)
             else:
                 valstr = val
             Logger.print2("| %15s | %15s |" % (key, valstr))
             vals.append(val)
         Logger.print2("-" * 37)
     return

コード例 #8

ファイル: logger.py プロジェクト: bulletphysics/bullet3

 def print_tabular(self):
   """
       Print all of the diagnostics from the current iteration
       """
   if (MPIUtil.is_root_proc()):
     vals = []
     Logger.print2("-" * 37)
     for key in self.log_headers:
       val = self.log_current_row.get(key, "")
       if isinstance(val, float):
         valstr = "%8.3g" % val
       elif isinstance(val, int):
         valstr = str(val)
       else:
         valstr = val
       Logger.print2("| %15s | %15s |" % (key, valstr))
       vals.append(val)
     Logger.print2("-" * 37)
   return

コード例 #9

ファイル: rl_agent.py プロジェクト: jiapei100/bullet3

 def _enable_int_output(self):
     return MPIUtil.is_root_proc() and self.int_output_dir != ""

コード例 #10

 def print2(str):
     if (MPIUtil.is_root_proc()):
         print(str)
     return

コード例 #11

ファイル: mpi_solver.py プロジェクト: bulletphysics/bullet3

 def _is_root(self):
   return MPIUtil.is_root_proc()

コード例 #12

ファイル: logger.py プロジェクト: bulletphysics/bullet3

 def print2(str):
   if (MPIUtil.is_root_proc()):
     print(str)
   return

コード例 #13

 def _enable_int_output(self):
     return MPIUtil.is_root_proc() and self.int_output_dir != ""

コード例 #14

 def _is_root(self):
     return MPIUtil.is_root_proc()