Fix typo

rosinality · Sep 22, 2020 · c96b675 · c96b675
1 parent 0ea71ec
commit c96b675
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/apply_factor.py b/apply_factor.py
@@ -67,8 +67,8 @@
 
     trunc = g.mean_latent(4096)
 
-    if torch_seed > 0:
-        torch.manual_seed(torch_seed)
+    if args.torch_seed > 0:
+        torch.manual_seed(args.torch_seed)
     latent = torch.randn(args.n_sample, 512, device=args.device)
     latent = g.get_latent(latent)