agents icon indicating copy to clipboard operation
agents copied to clipboard

How can we load from specific checkpoints use checkpointer?

Open wj210 opened this issue 3 years ago • 1 comments

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?

wj210 avatar Jan 12 '23 08:01 wj210

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

See policy_loader_test.py

sguada avatar Jan 12 '23 11:01 sguada