Update mymodels.py

mymodels.py

@@ -5,7 +5,7 @@ import cv2
 import numpy as np
 
 __all__ = [
-    'color_cluster','Color2Sketch', 'Sketch2Color', 'Discriminator',
+    'color_cluster', 'Color2Sketch', 'Sketch2Color', 'Discriminator',
 ]
 
 
@@ -259,7 +259,7 @@ class Color2Sketch(nn.Module):
         if pretrained:
             print('Loading pretrained {0} model...'.format('Color2Sketch'), end=' ')
             checkpoint = torch.load('color2edge.pth')
-            self.load_state_dict(checkpoint['netG'], strict=True)
+            self.load_state_dict(checkpoint['netG'], strict=False)
             print("Done!")
         else:
             self.apply(weights_init)