Skip to content
Snippets Groups Projects
Commit 5faf51b4 authored by Hsien-Chin Lin's avatar Hsien-Chin Lin
Browse files

wip

parent c9bacd6c
Branches
Tags
No related merge requests found
......@@ -239,17 +239,17 @@ if __name__ == '__main__':
policy_sys.log_dir = config_save_path.replace('configs', 'logs')
policy_sys.save_dir = save_path
# logging.info(f"Evaluating at start - {time_now}" + '-'*60)
# time_now = time.time()
# eval_dict = eval_policy(conf, policy_sys, env, sess,
# save_eval, log_save_path)
# logging.info(f"Finished evaluating, time spent: {time.time() - time_now}")
# for key in eval_dict:
# tb_writer.add_scalar(key, eval_dict[key], 0)
# best_complete_rate = eval_dict['complete_rate']
# best_success_rate = eval_dict['success_rate_strict']
# best_return = eval_dict['avg_return']
logging.info(f"Evaluating at start - {time_now}" + '-'*60)
time_now = time.time()
eval_dict = eval_policy(conf, policy_sys, env, sess,
save_eval, log_save_path)
logging.info(f"Finished evaluating, time spent: {time.time() - time_now}")
for key in eval_dict:
tb_writer.add_scalar(key, eval_dict[key], 0)
best_complete_rate = eval_dict['complete_rate']
best_success_rate = eval_dict['success_rate_strict']
best_return = eval_dict['avg_return']
logging.info("Start of Training: " +
time.strftime("%Y-%m-%d-%H-%M-%S", time.localtime()))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment