Update CUB-200-2011.py

PRIS-CV · Apr 21, 2020 · 1c1ff52 · 1c1ff52
1 parent 1524f5b
commit 1c1ff52
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/CUB-200-2011.py b/CUB-200-2011.py
@@ -273,7 +273,7 @@ def get_params():
     # Training settings
     parser = argparse.ArgumentParser(description='PyTorch MC2_AutoML Example')
 
-    parser.add_argument('--alpha_1', type=float, default=2.0, metavar='ALPHA',
+    parser.add_argument('--alpha_1', type=float, default=1.5, metavar='ALPHA',
                         help='alpha_1 value (default: 2.0)')
     parser.add_argument('--beta_1', type=float, default=20.0, metavar='BETA',
                         help='beta_1 value (default: 20.0)')