diff --git a/multi_run.py b/multi_run.py
index e8c9c67..2bb5ecd 100644
--- a/multi_run.py
+++ b/multi_run.py
@@ -26,25 +26,25 @@ if __name__ == '__main__':
             arg_dict.update(model_type=model)
             raw_conf = dict(data_speed_factor=0.0, data_speed_ratio=0.0, data_mask_ratio=0.0,
                             data_noise_ratio=0.0, data_shift_ratio=0.0, data_loudness_ratio=0.0,
-                            data_stretch=False)
+                            data_stretch=False, train_epochs=101)
             all_conf = dict(data_speed_factor=0.7, data_speed_ratio=0.2, data_mask_ratio=0.2,
                             data_noise_ratio=0.4, data_shift_ratio=0.4, data_loudness_ratio=0.4,
-                            data_stretch=True)
+                            data_stretch=True, train_epochs=51)
             speed_conf = dict(data_speed_factor=0.7, data_speed_ratio=0.2, data_mask_ratio=0.0,
                               data_noise_ratio=0.0, data_shift_ratio=0.0, data_loudness_ratio=0.0,
-                              data_stretch=True)
+                              data_stretch=True, train_epochs=51)
             mask_conf = dict(data_speed_factor=0.0, data_speed_ratio=0.0, data_mask_ratio=0.2,
                              data_noise_ratio=0.0, data_shift_ratio=0.0, data_loudness_ratio=0.0,
-                             data_stretch=True)
+                             data_stretch=True, train_epochs=51)
             noise_conf = dict(data_speed_factor=0.0, data_speed_ratio=0.0, data_mask_ratio=0.0,
                               data_noise_ratio=0.4, data_shift_ratio=0.0, data_loudness_ratio=0.0,
-                              data_stretch=True)
+                              data_stretch=True, train_epochs=51)
             shift_conf = dict(data_speed_factor=0.0, data_speed_ratio=0.0, data_mask_ratio=0.0,
                               data_noise_ratio=0.0, data_shift_ratio=0.4, data_loudness_ratio=0.0,
-                              data_stretch=True)
+                              data_stretch=True, train_epochs=51)
             loudness_conf = dict(data_speed_factor=0.0, data_speed_ratio=0.0, data_mask_ratio=0.0,
                                  data_noise_ratio=0.0, data_shift_ratio=0.0, data_loudness_ratio=0.4,
-                                 data_stretch=True)
+                                 data_stretch=True, train_epochs=51)
 
             for dicts in [raw_conf, all_conf, speed_conf, mask_conf,noise_conf, shift_conf, loudness_conf]: