def set_seed(seed): seed %= 4294967294 global seed_ seed_ = seed import random random.seed(seed) np.random.seed(seed) import torch torch.manual_seed(seed) # 为CPU设置随机种子 torch.cuda.manual_seed(seed) # 为[当前]GPU设置随机种子(不是所有GPU) print(colorize(f"using seed {seed}", "green"))
def set_seed(seed): seed %= 4294967294 global seed_ seed_ = seed import random random.seed(seed) np.random.seed(seed) import torch torch.manual_seed(seed) torch.cuda.manual_seed(seed) print(colorize(f"using seed {seed}", "green"))
def log(s, with_prefix=True, with_timestamp=True, color=None): if not _disabled: out = s if with_prefix and not _disable_prefix: out = _prefix_str + out if with_timestamp: now = datetime.datetime.now() # dateutil.tz.tzlocal()) timestamp = now.strftime('%Y-%m-%d %H:%M:%S.%f %Z') out = "%s | %s" % (timestamp, out) if color is not None: out = colorize(out, color) if not _log_tabular_only: # Also log to stdout print(out) for fd in list(_text_fds.values()): fd.write(out + '\n') fd.flush() sys.stdout.flush()
def _new_from_args(cls, parsed_args, *args, **params): silent = params.pop("_silent", False) args_info = _get_info(cls) prefix_ = _get_prefix(cls) # params = dict() for arg_name, arg_info in args_info.items(): prefixed_arg_name = prefix_ + arg_name if hasattr(parsed_args, prefixed_arg_name): val = getattr(parsed_args, prefixed_arg_name) if val is not None: if arg_info['mapper']: params[arg_name] = arg_info['mapper'](val) else: params[arg_name] = val if not silent: print( colorize( "using argument %s with value %s" % (arg_name, val), "yellow")) return cls(*args, **params)