diff --git a/mainscripts/Trainer.py b/mainscripts/Trainer.py index e48fb20..ea1eff5 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() ) ) @@ -174,13 +174,13 @@ def trainerThread (s2c, c2s, e, 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) + 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: