Add a CLI argument to seed the RNG

Objective: compare the effect of eigenvectors on the same samples.
rosinality · Sep 22, 2020 · c5d726c · c5d726c
1 parent 4c54835
commit c5d726c
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/apply_factor.py b/apply_factor.py
@@ -51,6 +51,12 @@
         type=str,
         help="name of the closed form factorization result factor file",
     )
+    parser.add_argument(
+        "--torch_seed",
+        type=int,
+        default=0,
+        help="seed for generating random numbers",
+    )
 
     args = parser.parse_args()
 
@@ -61,6 +67,7 @@
 
     trunc = g.mean_latent(4096)
 
+    torch.manual_seed(torch_seed)
     latent = torch.randn(args.n_sample, 512, device=args.device)
     latent = g.get_latent(latent)