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

Author: lucidrains

stylegan2_pytorch/stylegan2_pytorch.py

@@ -560,7 +560,7 @@ def __init__(self, image_size, network_capacity = 16, fq_layers = [], fq_dict_si
         num_init_filters = 3 if not transparent else 4
 
         blocks = []
-        filters = [num_init_filters] + [(64) * (2 ** i) for i in range(num_layers + 1)]
+        filters = [num_init_filters] + [(network_capacity * 4) * (2 ** i) for i in range(num_layers + 1)]
 
         set_fmap_max = partial(min, fmap_max)
         filters = list(map(set_fmap_max, filters))