DenseNet161 with SGD

ajing · Jun 15, 2017 · e7c397a · e7c397a
1 parent 706d5c7
commit e7c397a
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/util.py b/util.py
@@ -222,9 +222,6 @@ def save_time(self, start_time, end_time):
 
 
 if __name__ == '__main__':
-    # a = np.random.randn(100, 17)
-    # np.savetxt('probs/model_1.txt', a)
-    # optimize_threshold(['probs/model_1.txt'], 'data')
     validation = KgForestDataset(
         split='validation-3000',
         transform=Compose(
@@ -239,6 +236,6 @@ def save_time(self, start_time, end_time):
         height=256,
         width=256
     )
-    dataloader = DataLoader(validation)
-    save_results([resnet18_planet, resnet34_planet, resnet50_planet,
-        densenet121, densenet169, densenet161,], dataloader)
+    files = ['densenet121.txt', 'densenet161.txt', 'densenet169.txt', 'resnet18_planet.txt',
+             'resnet34_planet.txt', 'resnet50_planet.txt']
+    optimize_threshold(files, resolution=500, labels=validation.labels)