bug fix

backend/disentangle_concepts.py

@@ -92,7 +92,7 @@ def regenerate_images(model, z, decision_boundary, min_epsilon=-3, max_epsilon=3
         
         if layers:
             W_f = torch.empty_like(W).copy_(W)
-            W_f[layers] = W_0[layers]
+            W_f[:, layers, :] = W_0[:, layers, :]
             img = G.synthesis(W_f, noise_mode='const')
         else:
             img = G.synthesis(W_0, noise_mode='const')