34 lines
740 B
Python
34 lines
740 B
Python
import argparse
|
|
import numpy as np
|
|
import pandas as pd
|
|
import torch
|
|
import math
|
|
import random
|
|
import os
|
|
import sys
|
|
import datetime
|
|
import copy
|
|
import time
|
|
|
|
|
|
def set_random_seed(args):
|
|
seed = args.seed
|
|
torch.manual_seed(seed)
|
|
torch.cuda.manual_seed_all(seed)
|
|
torch.backends.cudnn.deterministic = True
|
|
torch.backends.cudnn.benchmark = False
|
|
np.random.seed(seed)
|
|
random.seed(seed)
|
|
os.environ["PYTHONHASHSEED"] = str(seed)
|
|
|
|
|
|
def str2bool(v):
|
|
if isinstance(v, bool):
|
|
return v
|
|
if v.lower() in ("yes", "true", "t", "y", "1"):
|
|
return True
|
|
elif v.lower() in ("no", "false", "f", "n", "0"):
|
|
return False
|
|
else:
|
|
raise argparse.ArgumentTypeError("Boolean value expected.")
|