Checkpoint creation

[SabbirAhmad26]

Hi
I hope you are doing well. I am wondering how do you generate these checkpoints? either .npz files or the recent ones in best_new_checkpoints?. I did some training but the format of the new checkpoint is different and I am having trouble using new_viz.py or viz_from_checkpoint. I would be grateful if you can guide me in this matter.

[pengzhenghao]

Can you try this?

https://github.com/decisionforce/CoPO/blob/main/copo_code/copo/new_vis.py

[EhsanSabouni]

Hi Thanks for you response.

This the error I will get.

Traceback (most recent call last):
File "/home/rastic-admin/Documents/GitHub/CoPO-main(2)/CoPO-main/copo_code/copo/eval/get_policy_function_from_checkpoint.py", line 23, in get_policy_function_from_checkpoint
unpickled = pickle.loads(data)
ValueError: unsupported pickle protocol: 5

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File "/home/rastic-admin/Documents/GitHub/CoPO-main(2)/CoPO-main/copo_code/copo/new_vis.py", line 83, in
policy_function = get_policy_function_from_checkpoint(algo=algo, ckpt=ckpt_path)
File "/home/rastic-admin/Documents/GitHub/CoPO-main(2)/CoPO-main/copo_code/copo/eval/get_policy_function_from_checkpoint.py", line 28, in get_policy_function_from_checkpoint
worker = pickle.loads(unpickled.pop("worker"))
TypeError: a bytes-like object is required, not 'dict'.

The files that I get in the checkpoint is .pkl and metadata files. The problem is I do not know the format of the files that are compatible with new_viz.py so I can generate them. I mean the format of the files stored in new_best_checkpoints folder.