arg defaults changed

Author: gaborvecsei

fed_learn/args_helper.py

@@ -7,16 +7,16 @@ def get_args():
     parser.add_argument("-n", "--name", help="Name of the experiment", type=str, required=True)
     parser.add_argument("-oe", "--overwrite-experiment", help="Overwrite existing experiment", action="store_true",
                         required=False)
-    parser.add_argument("-e", "--global-epochs", help="Number of global (server) epochs", type=int, default=10,
+    parser.add_argument("-e", "--global-epochs", help="Number of global (server) epochs", type=int, default=1000,
                         required=False)
     parser.add_argument("-c", "--clients", help="Number of clients", type=int, default=100, required=False)
-    parser.add_argument("-f", "--fraction", help="Client fraction to use", type=float, default=0.2,
+    parser.add_argument("-f", "--fraction", help="Client fraction to use", type=float, default=0.1,
                         required=False)
     parser.add_argument("-d", "--debug", help="Debugging", action="store_true", required=False)
 
-    parser.add_argument("-lr", "--learning-rate", help="Learning rate", type=float, default=0.01, required=False)
+    parser.add_argument("-lr", "--learning-rate", help="Learning rate", type=float, default=0.15, required=False)
     parser.add_argument("-b", "--batch-size", help="Batch Size", type=int, default=32, required=False)
-    parser.add_argument("-ce", "--client-epochs", help="Number of epochs for the clients", type=int, default=5,
+    parser.add_argument("-ce", "--client-epochs", help="Number of epochs for the clients", type=int, default=1,
                         required=False)
     args = parser.parse_args()
     return args