diff --git a/run_with_oversampling.py b/run_with_oversampling.py
index b21e5e6f98df00a5f866c10db280f29e3366b014..023a8a6f3019fc24ba8374ac780931570c95f4e8 100644
--- a/run_with_oversampling.py
+++ b/run_with_oversampling.py
@@ -26,8 +26,9 @@ def load_stations(external_station_list):
 
 def main(parser_args):
     plots = remove_items(DEFAULT_PLOT_LIST, "PlotConditionalQuantiles")
-    workflow = DefaultWorkflow(stations=load_stations('supplement/German_background_station.json')[:75],
+    workflow = DefaultWorkflow(stations=load_stations('supplement/German_background_station.json'),
         #stations=["DEBW013", "DEBW087", "DEBW107", "DEBW076"],
+        epochs=150,
         train_model=True, create_new_model=True, network="UBA",
         model=IntelliO3_ts_architecture, oversampling_method="bin_oversampling",
         evaluate_bootstraps=False,  # plot_list=["PlotCompetitiveSkillScore"],
diff --git a/run_without_oversampling.py b/run_without_oversampling.py
index 5b51ffa0d73c0a92491f0cf52285dd53dcec5cd3..bda899ca586093c2456488c94bc54b6784d9ac6c 100644
--- a/run_without_oversampling.py
+++ b/run_without_oversampling.py
@@ -26,8 +26,9 @@ def load_stations(external_station_list = None):
 
 def main(parser_args):
     plots = remove_items(DEFAULT_PLOT_LIST, "PlotConditionalQuantiles")
-    workflow = DefaultWorkflow(stations=load_stations('supplement/German_background_station.json')[:75],
+    workflow = DefaultWorkflow(stations=load_stations('supplement/German_background_station.json'),
         #stations=["DEBW013", "DEBW087", "DEBW107", "DEBW076"],
+        epochs=150,
         train_model=True, create_new_model=True, network="UBA",
         model=IntelliO3_ts_architecture,
         evaluate_bootstraps=False,  # plot_list=["PlotCompetitiveSkillScore"],