Skip to content

Commit

Permalink
Dev (#11)
Browse files Browse the repository at this point in the history
* 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

* fix: a bug of sorting in torch-version checkpoint loading
  • Loading branch information
xionghuichen committed Nov 11, 2022
1 parent ee41618 commit 28bedd0
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions RLA/easy_log/tester.py
Original file line number Diff line number Diff line change
Expand Up @@ -695,6 +695,7 @@ 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):
Expand Down

0 comments on commit 28bedd0

Please sign in to comment.