| import random | |
| import numpy as np | |
| import torch | |
| def fix_random(seed: int = 0) -> None: | |
| np.random.seed(seed) | |
| torch.manual_seed(seed) | |
| torch.cuda.manual_seed(seed) | |
| torch.cuda.manual_seed_all(seed) | |
| torch.backends.cudnn.deterministic = True | |
| torch.backends.cudnn.benchmark = False | |
| def randid(k: int = 4): | |
| charset = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789' | |
| return ''.join(random.choices(charset, k=k)) | |