diff --git a/train.py b/train.py index 4dcf598..874c80f 100644 --- a/train.py +++ b/train.py @@ -69,8 +69,8 @@ def setup_seed(seed): torch.manual_seed(seed) torch.cuda.manual_seed_all(seed) -# np.random.seed(seed) -# random.seed(seed) + # np.random.seed(seed) + # random.seed(seed) torch.backends.cudnn.deterministic = True