diff --git a/networks.py b/networks.py
index 25b34abd..4653a063 100644
--- a/networks.py
+++ b/networks.py
@@ -20,7 +20,7 @@ def assign_adain_params(adain_params, model):
             std = adain_params[:, m.num_features:2*m.num_features]
             m.bias = mean.contiguous().view(-1)
             m.weight = std.contiguous().view(-1)
-            if adain_params.size(1) > 2*m.num_features:
+            if adain_params.size(1) >= 2*m.num_features:
                 adain_params = adain_params[:, 2*m.num_features:]