changed default train config

Author: harsha-simhadri

examples/pytorch/FastCells/train_classifier.py

@@ -533,8 +533,7 @@ def str2bool(v):
 
 
 if __name__ == '__main__':
-    config = TrainingConfig()
-    parser = argparse.ArgumentParser("train a GRU based neural network for keyword spotting")
+    parser = argparse.ArgumentParser("train a RNN based neural network for keyword spotting")
 
     # all the training parameters
     parser.add_argument("--epochs", help="Number of epochs to train", type=int)
@@ -587,6 +586,7 @@ def str2bool(v):
                         action="store_true")
     args = parser.parse_args()
 
+    config = TrainingConfig()
     if args.config:
         config.load(args.config)
 

examples/pytorch/FastCells/training_config.py

@@ -53,14 +53,14 @@ def __init__(self):
 class TrainingOptions:
     def __init__(self):
         self.max_epochs = 30
-        self.learning_rate = 1e-3
+        self.learning_rate = 1e-2
         self.lr_scheduler = None
         self.lr_peaks = 1
         self.lr_min = 1e-5
         self.lr_gamma = 1
         self.lr_step_size = 1
         self.batch_size = 128
-        self.optimizer = "RMSprop"
+        self.optimizer = "Adam"
         self.optimizer_options = OptimizerOptions()
         self.use_gpu = False
         self.rolling = False