Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How can we load from specific checkpoints use checkpointer? #817

Open
wj210 opened this issue Jan 12, 2023 · 1 comment
Open

How can we load from specific checkpoints use checkpointer? #817

wj210 opened this issue Jan 12, 2023 · 1 comment

Comments

@wj210
Copy link

wj210 commented Jan 12, 2023

I save the model during certain timesteps so at the end of the run i would have multiple checkpoints,
the files are saved as ckpt-1000, ckpt-2000 ...

If for example, i would like to load the files and evaluate the agent with the loaded variables for each checkpoint, how do i go about?
The initialise_or_restore function doesnt seem to take in any file names.

Is there a way to do this?

@sguada
Copy link
Member

sguada commented Jan 12, 2023

You can use policy_loader.load() to load a saved policy and evaluate it.

See policy_loader_test.py

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants