diff --git a/apply_factor.py b/apply_factor.py index 80a3b3a2..f81726c7 100755 --- 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, )