added seeding

cikm_inter_dst_predrnn_run_taasss.py

@@ -6,6 +6,8 @@ import numpy as np
 from tqdm import tqdm
 import os
 import sys
+import torch
+import random
 
 import core.trainer as trainer
 from cikm_inter_dst_predrnn_run_taasss_utils import get_batcher, padding_taasss, change_taasss_dims
@@ -67,6 +69,15 @@ batch_size = args.batch_size
 #     args.patch_size = 16
 
 
+def seed_everything(seed):
+    random.seed(seed)
+    os.environ['PYTHONHASHSEED'] = str(seed)
+    np.random.seed(seed)
+    torch.manual_seed(seed)
+    torch.cuda.manual_seed(seed)
+    torch.backends.cudnn.deterministic = True
+
+
 def schedule_sampling(eta: float, itr: int):
     """
     Return
@@ -158,5 +169,6 @@ def wrapper_train(model: Model):
 
 
 print("Initializing models", file=sys.stderr)
+seed_everything(42)
 model = Model(args)
 wrapper_train(model)