Update network.py

network.py

@@ -104,7 +104,7 @@ class Style_Transfer_Network(nn.Module):
     # encode style images
     style_features = []
     for style in styles:
-        if preserve_color == 'whiten_and_color' or preserve_color == 'histogram_matching':
+        if preserve_color == 'whitening_and_coloring' or preserve_color == 'histogram_matching':
                 style = batch_wct(style, content)
         style_features.append(self.encoder(style))
 
@@ -117,7 +117,7 @@ class Style_Transfer_Network(nn.Module):
     transformed_feature = sum(transformed_features)
 
     stylized_image = self.decoder(transformed_feature)
-    if preserve_color == "whiten_and_color":
+    if preserve_color == "whitening_and_coloring":
         stylized_image = batch_wct(stylized_image, content)
     if preserve_color == "histogram_matching":
         stylized_image = batch_histogram_matching(stylized_image, content)