Update mymodels.py

mymodels.py

@@ -258,7 +258,6 @@ class Color2Sketch(nn.Module):
         self.decoder = Decoder()
         if pretrained:
             print('Loading pretrained {0} model...'.format('Color2Sketch'), end=' ')
-            assert os.path.isdir('checkpoint'), 'Error: no checkpoint directory found!'
             checkpoint = torch.load('color2edge.pth')
             self.load_state_dict(checkpoint['netG'], strict=True)
             print("Done!")
@@ -396,7 +395,6 @@ class Sketch2Color(nn.Module):
         self.decoder = Decoder()
         if pretrained:
             print('Loading pretrained {0} model...'.format('Sketch2Color'), end=' ')
-            assert os.path.isdir('checkpoint'), 'Error: no checkpoint directory found!'
             checkpoint = torch.load('edge2color.pth')
             self.load_state_dict(checkpoint['netG'], strict=True)
             print("Done!")