From 6296c1bef817996d500c661729f5b25db4f1e184 Mon Sep 17 00:00:00 2001 From: Xiong-Hui Chen Date: Thu, 11 Aug 2022 01:14:13 +0800 Subject: [PATCH] Dev (#12) * Update README.md * Dev (#20) * fix: fix bugs of torch-version ckp loader * refactor: add sync_timestep for hp loader * fix: minor changes for version compatibility * Dev (#21) * fix: fix bugs of torch-version ckp loader * refactor: add sync_timestep for hp loader * fix: minor changes for version compatibility * fix: a bug of sorting in torch-version checkpoint loading * Dev (#22) * fix: fix bugs of torch-version ckp loader * refactor: add sync_timestep for hp loader * fix: minor changes for version compatibility * fix: a bug of sorting in torch-version checkpoint loading * refactor: robust multi-key plot implementation * feat: supoort pretty plotter * refactor(log plotter): record scores of the log plotter * fix(exp_loader): add parameter ckp_index * update readme * Dev (#23) * fix: fix bugs of torch-version ckp loader * refactor: add sync_timestep for hp loader * fix: minor changes for version compatibility * fix: a bug of sorting in torch-version checkpoint loading * refactor: robust multi-key plot implementation * feat: supoort pretty plotter * refactor(log plotter): record scores of the log plotter * fix(exp_loader): add parameter ckp_index * update readme * rm unsolved merge * feat: tf-v2 compatible * refactor: add timestep recorder. refactor on exp_loader * test: add test data * feat(plot): track the hyper-parameter from the exp_manager instead of the experiment name. refactor the plot_func for better readability --- RLA/easy_log/exp_loader.py | 5 ++--- RLA/easy_log/tester.py | 2 +- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/RLA/easy_log/exp_loader.py b/RLA/easy_log/exp_loader.py index c2206b1..56f7c4a 100644 --- a/RLA/easy_log/exp_loader.py +++ b/RLA/easy_log/exp_loader.py @@ -82,9 +82,8 @@ def load_from_record_date(self, var_prefix: Optional[str] = None, variable_list: """ if self.is_valid_config: loaded_tester = Tester.load_tester(self.load_date, self.task_name, self.data_root) - if verbose: - print("attrs of the loaded tester") - pprint(loaded_tester.__dict__) + print("attrs of the loaded tester") + pprint(loaded_tester.__dict__) # load checkpoint load_res = {} if var_prefix is not None: diff --git a/RLA/easy_log/tester.py b/RLA/easy_log/tester.py index f25b472..66f7807 100644 --- a/RLA/easy_log/tester.py +++ b/RLA/easy_log/tester.py @@ -15,6 +15,7 @@ import datetime import os.path as osp import pprint +import numpy as np from RLA.easy_log.time_step import time_step_holder @@ -695,7 +696,6 @@ def load_checkpoint(self, ckp_index=None): pprint.pprint(all_ckps) if ckp_index is None: ckp_index = all_ckps[-1].split('checkpoint-')[1].split('.pt')[0] - print("loaded checkpoints:", "checkpoint-{}.pt".format(ckp_index)) return ckp_index, torch.load(self.checkpoint_dir + "checkpoint-{}.pt".format(ckp_index)) def auto_parse_info(self):