From a33fa7ddebd662124908cb9ea15c71af8821fad3 Mon Sep 17 00:00:00 2001 From: leno3003 Date: Thu, 31 Mar 2022 10:04:38 +0200 Subject: [PATCH] added sumbmodules --- mainscripts/Trainer.py | 34 +++++++++++++++++----------------- 1 file changed, 17 insertions(+), 17 deletions(-) diff --git a/mainscripts/Trainer.py b/mainscripts/Trainer.py index ea1eff5..8c543c4 100644 --- a/mainscripts/Trainer.py +++ b/mainscripts/Trainer.py @@ -84,15 +84,15 @@ def trainerThread (s2c, c2s, e, if model.get_target_iter() != 0: if is_reached_goal: - answer_filename = 'workspace/interact/interact_dict.pkl' - if os.path.exists(answer_filename): - with open(answer_filename, 'rb') as file: - parameters = pickle.load(file) - if parameters['NoInteractiveMode']: - print("Entro 1") - model_save() - os._exit(1) - else: +# answer_filename = 'workspace/interact/interact_dict.pkl' +# if os.path.exists(answer_filename): +# with open(answer_filename, 'rb') as file: +# parameters = pickle.load(file) +# if parameters['NoInteractiveMode']: +# print("Entro 1") +# model_save() +# os._exit(1) +# else: io.log_info('Model already trained to target iteration. You can use preview.') else: io.log_info('Starting. Target iteration: %d. Press "Enter" to stop training and save model.' % ( model.get_target_iter() ) ) @@ -173,14 +173,14 @@ def trainerThread (s2c, c2s, e, model_save() is_reached_goal = True io.log_info ('You can use preview now.') - answer_filename = 'workspace/interact/interact_dict.pkl' - if os.path.exists(answer_filename): - with open(answer_filename, 'rb') as file: - parameters = pickle.load(file) - if parameters['NoInteractiveMode']: - print("Entro 2 ") - model_save() - os._exit(1) + #answer_filename = 'workspace/interact/interact_dict.pkl' + #if os.path.exists(answer_filename): + # with open(answer_filename, 'rb') as file: + # parameters = pickle.load(file) + # if parameters['NoInteractiveMode']: + # print("Entro 2 ") + # model_save() + # os._exit(1) need_save = False while time.time() - last_save_time >= save_interval_min*60: