Update train_model.py

scripts/train_model.py

@@ -27,13 +27,13 @@ exp_config = ExperimentConfiguration({'n_augment': 1,
                                       'n_epochs': 500,
                                       'n_iters': 1,
                                       'batch_size': 512,
-                                      'learning_rate': 0.0001,
+                                      'learning_rate': 0.0001,  # may use 0.001 for model 'left_center_right'
                                       'model_name': 'one_sided',
                                       # possible values: {'default', 'left_center_right', 'one_sided'}
                                       'model_config': {
                                           'random_scale': {
-                                              'minval': -10,
-                                              'maxval': 10,
+                                              'minval': -1,
+                                              'maxval': 1,
                                           },
                                           'random_shift': {
                                               'minval': 0,