spirl
spirl copied to clipboard
cannot run the command of Skill Space Policy w/o prior:
When I run the baseline command of Skill Space Policy w/o prior:python spirl/rl/train.py --path=spirl/configs/hrl/kitchen/no_prior/ --seed=0 --prefix=SSP_noPrior_kitchen_seed0
it reports the following errors:
Loading from: ./experiments/skill_prior_learning/kitchen/hierarchical/weights
Traceback (most recent call last):
File "/data/users/siqi_li/spirl/spirl/rl/train.py", line 325, in <module>
RLTrainer(args=get_args())
File "/data/users/siqi_li/spirl/spirl/rl/train.py", line 61, in __init__
self.agent = self._hp.agent(self.conf.agent)
File "/data/users/siqi_li/spirl/spirl/rl/components/agent.py", line 326, in __init__
super().__init__(config)
File "/data/users/siqi_li/spirl/spirl/rl/components/agent.py", line 212, in __init__
self.ll_agent = self._hp.ll_agent(self._hp.overwrite(self._hp.ll_agent_params))
File "/data/users/siqi_li/spirl/spirl/rl/agents/skill_space_agent.py", line 18, in __init__
self.load_model_weights(self._policy, self._hp.model_checkpoint, self._hp.model_epoch)
File "/data/users/siqi_li/spirl/spirl/rl/components/agent.py", line 160, in load_model_weights
checkpoint_path = CheckpointHandler.get_resume_ckpt_file(epoch, checkpoint_dir)
File "/data/users/siqi_li/spirl/spirl/components/checkpointer.py", line 30, in get_resume_ckpt_file
max_epoch = np.max(CheckpointHandler.get_epochs(path))
File "/data/users/siqi_li/spirl/spirl/components/checkpointer.py", line 20, in get_epochs
raise ValueError("No checkpoints found at {}!".format(path))
ValueError: No checkpoints found at ./experiments/skill_prior_learning/kitchen/hierarchical/weights!
I don't know why this happened. It seems the hierarchical model weight is not given, and the pre-trained skill prior models are given in readme. Could you please help me about this problem?