Fix typo

rosinality · Nov 6, 2023 · 5fa9f30 · 5fa9f30
1 parent 25b87a0
commit 5fa9f30
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/apply_factor.py b/apply_factor.py
@@ -75,7 +75,7 @@
 
     img_dict = dict()
 
-    for u in torch.linspace(- args.degree, args.degree, args.d_num):
+    for u in torch.linspace(- args.degree, args.degree, args.degree_num):
 
         img_batch, _ = g(
             [latent + u * direction],
@@ -103,5 +103,5 @@
         f"{args.out_prefix}_index-{args.index}_degree-{args.degree}.png",
         normalize=True,
         range=(-1, 1),
-        nrow=args.d_num,
+        nrow=args.degree_num,
     )