From 11cf29197434d3bf2addd57621c8c41794d992dd Mon Sep 17 00:00:00 2001 From: Pavel Date: Wed, 27 Nov 2019 10:56:39 +0300 Subject: [PATCH] Fixed adain parameter slicing. --- networks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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:]