From f75c4730f53bab7c18b2a811f01fe8158dec655b Mon Sep 17 00:00:00 2001 From: Henry Tu Date: Tue, 7 Jun 2022 10:41:39 -0400 Subject: [PATCH] Set seed in main function, instead of beginning of execution --- examples/ltc_backend_bert.py | 2 -- examples/ltc_backend_mnist.py | 2 -- 2 files changed, 4 deletions(-) diff --git a/examples/ltc_backend_bert.py b/examples/ltc_backend_bert.py index 31cfe156e69..d309ba87136 100644 --- 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 index 7bb2db16deb..65f8a4e5f37 100644 --- 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",