diff --git a/model.py b/model.py index 6c6c87b..035efd0 100644 --- a/model.py +++ b/model.py @@ -6,7 +6,7 @@ def model_static(pretrained=False, **kwargs): model = ResNet([3, 4, 6, 3], **kwargs) if pretrained: - print 'loading saved model weights' + print('loading saved model weights') model_dict = model.state_dict() snapshot = torch.load(pretrained) snapshot = {k: v for k, v in snapshot.items() if k in model_dict}