diff --git a/run.py b/run.py index 8e4d9c46f4a39224335dd65b689f519943166d0f..556cd0b59ed023178fa7e6df1b5b03b9f6c5f157 100644 --- a/run.py +++ b/run.py @@ -17,7 +17,7 @@ def main(parser_args): with RunEnvironment(): ExperimentSetup(parser_args, stations=['DEBW107', 'DEBY081', 'DEBW013', 'DEBW076', 'DEBW087', 'DEBW001'], - station_type='background', trainable=True) + station_type='background', trainable=False, create_new_model=False) PreProcessing() ModelSetup() diff --git a/src/run_modules/model_setup.py b/src/run_modules/model_setup.py index e3945a542d60b09dc9855bd28be87cdba729ed72..d2c8d93fc957ecb2990e99000cbd3588e2d83eef 100644 --- a/src/run_modules/model_setup.py +++ b/src/run_modules/model_setup.py @@ -10,8 +10,8 @@ import tensorflow as tf from src.model_modules.keras_extensions import HistoryAdvanced, CallbackHandler # from src.model_modules.model_class import MyBranchedModel as MyModel -# from src.model_modules.model_class import MyLittleModel as MyModel -from src.model_modules.model_class import MyTowerModel as MyModel +from src.model_modules.model_class import MyLittleModel as MyModel +# from src.model_modules.model_class import MyTowerModel as MyModel from src.run_modules.run_environment import RunEnvironment