Update densenet_withglam.py

densenet_withglam.py

@@ -77,7 +77,7 @@ class GLAM(nn.Module):
         return out
 def get_model_with_attention(model_name, num_classes):
     if model_name == 'densenet169':
-        model = models.densenet169(pretrained=None)
+        model = models.densenet169(pretrained=False)
         in_channels = model.classifier.in_features
         model.features = nn.Sequential(model.features, nn.ReLU(inplace=True), GLAM(in_channels))
         model.classifier = nn.Linear(in_channels, num_classes)