From 6d39cf4536bdf0a4b117f83d5440bb4a4e194166 Mon Sep 17 00:00:00 2001
From: "v.gramlich1" <v.gramlichfz-juelich.de>
Date: Thu, 2 Sep 2021 08:51:01 +0200
Subject: [PATCH] Add imports to run_with_finetuning.py
IntelliO3_ts_architecture_finetune_all_dense,
IntelliO3_ts_architecture_finetune_outputs and
IntelliO3_ts_architecture_finetune_main_output
---
run_with_finetuning.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/run_with_finetuning.py b/run_with_finetuning.py
index 65611793..fc334a5c 100644
--- a/run_with_finetuning.py
+++ b/run_with_finetuning.py
@@ -5,7 +5,8 @@ import argparse
from mlair.workflows import DefaultWorkflow
from mlair.helpers import remove_items
from mlair.configuration.defaults import DEFAULT_PLOT_LIST
-from mlair.model_modules.model_class import IntelliO3_ts_architecture, IntelliO3_ts_architecture_freeze
+from mlair.model_modules.model_class import IntelliO3_ts_architecture, IntelliO3_ts_architecture_finetune_all_dense, \
+ IntelliO3_ts_architecture_finetune_outputs, IntelliO3_ts_architecture_finetune_main_output
import os
@@ -27,7 +28,7 @@ def main(parser_args):
#stations=["DEBW013", "DEBW087"],
epochs=1, external_weights="/home/vincentgramlich/mlair/data/weights/testrun_network_daily_model-best.h5",
train_model=True, create_new_model=True, network="UBA",
- model=IntelliO3_ts_architecture_freeze,
+ model=IntelliO3_ts_architecture_finetune_all_dense,
window_lead_time=1,
#oversampling_method="bin_oversampling", oversampling_bins=10, oversampling_rates_cap=100, window_lead_time=2,
evaluate_bootstraps=False, plot_list=["PlotContingency"],
--
GitLab