Merge branch 'master' of https://github.com/rosinality/stylegan2-pytorch

ppl.py

@@ -54,6 +54,12 @@ def lerp(a, b, t):
     parser.add_argument(
         "--crop", action="store_true", help="apply center crop to the images"
     )
+    parser.add_argument(
+        "--sampling",
+        default="end",
+        choices=["end", "full"],
+        help="set endpoint sampling method",
+    )
     parser.add_argument(
         "ckpt", metavar="CHECKPOINT", help="path to the model checkpoints"
     )
@@ -83,7 +89,10 @@ def lerp(a, b, t):
             noise = g.make_noise()
 
             inputs = torch.randn([batch * 2, latent_dim], device=device)
-            lerp_t = torch.rand(batch, device=device)
+            if args.sampling == "full":
+                lerp_t = torch.rand(batch, device=device)
+            else:
+                lerp_t = torch.zeros(batch, device=device)
 
             if args.space == "w":
                 latent = g.get_latent(inputs)