update

dyabel · Aug 6, 2022 · c3bdf85 · c3bdf85
1 parent 14e859d
commit c3bdf85
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tools/train.py b/tools/train.py
@@ -42,12 +42,12 @@ def train_model(cfg, local_rank, distributed, seed=214):
     np.random.seed(local_seed)
     random.seed(local_seed)
     #TODO
-    """
+    # """
     g = torch.Generator()
     g.manual_seed(local_seed)
     torch.backends.cudnn.benchmark = False
     torch.backends.cudnn.deterministic=True
-    """
+    # """
     model = build_zsl_pipeline(cfg)
     device = torch.device(cfg.MODEL.DEVICE)
     model = model.to(device)