Skip to content
Snippets Groups Projects

Stop Downloading Densenet in nosetests

Merged Anjith GEORGE requested to merge densenet into master
1 file
+ 1
1
Compare changes
  • Side-by-side
  • Inline
@@ -41,7 +41,7 @@ class DeepPixBiSExtractor(Extractor):
@@ -41,7 +41,7 @@ class DeepPixBiSExtractor(Extractor):
# model
# model
self.transforms = transforms
self.transforms = transforms
self.network = DeepPixBiS(pretrained=True)
self.network = DeepPixBiS(pretrained=False)
self.scoring_method = scoring_method
self.scoring_method = scoring_method
self.available_scoring_methods=['pixel_mean','binary','combined']
self.available_scoring_methods=['pixel_mean','binary','combined']
Loading