Fix typo

rosinality · Sep 19, 2020 · ec8ac0c · ec8ac0c
1 parent eb53d25
commit ec8ac0c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/apply_factor.py b/apply_factor.py
@@ -69,7 +69,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],
@@ -97,5 +97,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,
     )