Pad integers in filenames for fixed width

rosinality · Sep 18, 2020 · 8473784 · 8473784
1 parent 7be6e64
commit 8473784
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/apply_factor.py b/apply_factor.py
@@ -81,7 +81,7 @@
 
     grid = utils.save_image(
         torch.cat([img1, img, img2], 0),
-        f"{args.out_prefix}_index-{args.index}_degree-{args.degree}.png",
+        f"{args.out_prefix}_index-{args.index:02d}_degree-{args.degree}.png",
         normalize=True,
         range=(-1, 1),
         nrow=args.n_sample,