scheduler changed

fujenchu · Dec 4, 2017 · 3074da0 · 3074da0
1 parent 457e66a
commit 3074da0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/config.py b/config.py
@@ -32,7 +32,7 @@
 parser.add_argument('--num_query', type=int, default=15)
 parser.add_argument('--num_workers', type=int, default=24)
 parser.add_argument('--learning_rate', type=float, default=0.001)
-parser.add_argument('--drop_prob', type=float, default=0.2)
+parser.add_argument('--drop_prob', type=float, default=0.0)
 
 
 parser.add_argument('--log_step', type=int, default=100,

diff --git a/demo.py b/demo.py
@@ -64,7 +64,7 @@ def main(args):
                                           miniImageNetPath='/home/fujenchu/projects/dataset/miniImageNet_Ravi/',
                                           imgPerCls=600)
 
-    scheduler = StepLR(optimizer, step_size=10, gamma=0.1)
+    scheduler = StepLR(optimizer, step_size=40, gamma=0.5)
     ''' training'''
     for epoch in range(1000):
         scheduler.step()