Set seed in main function, instead of beginning of execution

llvm · Jun 7, 2022 · f75c473 · f75c473
1 parent d00f8c7
commit f75c473
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 4 deletions.
diff --git a/examples/ltc_backend_bert.py b/examples/ltc_backend_bert.py
@@ -126,8 +126,6 @@ def main(device='lazy', full_size=False):
 
 
 if __name__ == "__main__":
-    torch.manual_seed(0)
-
     parser = argparse.ArgumentParser()
     parser.add_argument(
         "-d",

diff --git a/examples/ltc_backend_mnist.py b/examples/ltc_backend_mnist.py
@@ -78,8 +78,6 @@ def forward(self, x):
 
 
 if __name__ == "__main__":
-    torch.manual_seed(0)
-
     parser = argparse.ArgumentParser()
     parser.add_argument(
         "-d",