From fe883840d49178d25a9d26e9068bb77af37f0309 Mon Sep 17 00:00:00 2001
From: lzzyzlbb <287246233@qq.com>
Date: Mon, 25 Oct 2021 19:45:06 +0800
Subject: [PATCH] fix stylegan path_batch_shrink type (#449)

---
 ppgan/models/styleganv2_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ppgan/models/styleganv2_model.py b/ppgan/models/styleganv2_model.py
index 827e6ed..e41c552 100644
--- a/ppgan/models/styleganv2_model.py
+++ b/ppgan/models/styleganv2_model.py
@@ -258,7 +258,7 @@ class StyleGAN2Model(BaseModel):
 
         l_g_total += l_g
         if current_iter % self.gen_iters == 0:
-            path_batch_size = max(1, batch // self.path_batch_shrink)
+            path_batch_size = max(1, int(batch // self.path_batch_shrink))
             noise = self.mixing_noise(path_batch_size, self.mixing_prob)
             fake_img, latents = self.nets['gen'](noise, return_latents=True)
             l_g_path, path_lengths, self.mean_path_length = g_path_regularize(
-- 
GitLab