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

@@ -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:

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):