Pad integers in filenames for fixed width

woctezuma · woctezuma · commit 4172c52a4592 · 2023-11-06T20:33:14.000+01:00
diff --git a/apply_factor.py b/apply_factor.py
@@ -87,7 +87,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,
