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)
