diff --git a/ppgan/models/styleganv2_model.py b/ppgan/models/styleganv2_model.py
index 827e6edc53b91948741f5da10c2257f4f7340e15..e41c55216182cee7cba216050a94fa4493b24e3e 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(