spirl icon indicating copy to clipboard operation
spirl copied to clipboard

cannot run the command of Skill Space Policy w/o prior:

Open abyssalfishLSQ opened this issue 10 months ago • 0 comments

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?

abyssalfishLSQ avatar Feb 26 '25 05:02 abyssalfishLSQ