DLR-RM / rl-baselines3-zoo

A training framework for Stable Baselines3 reinforcement learning agents, with hyperparameter optimization and pre-trained agents included.
https://rl-baselines3-zoo.readthedocs.io
MIT License
1.89k stars 494 forks source link

[Question] You must pass an environment when using `HerReplayBuffer` #432

Closed moneypi closed 5 months ago

moneypi commented 5 months ago

❓ Question

I try to train parking-v0 with command python train.py --algo tqc --env parking-v0 and everything seems fine. 20240118-152847 and I try to run python enjoy.py --algo tqc --env parking-v0 --folder logs/tqc/parking-v0_1/, I get errors as follow, 20240118-153105

and I try to fix it by

cd logs/tqc/parking-v0_1
mkdir tqc
cp best_model.zip tqc/parking-v0.zip

and when I ran python enjoy.py --algo tqc --env parking-v0 --folder logs/tqc/parking-v0_1/ again, I get errors as follow

(base) zhuqinghua@lcwt-DSS8440:~/workspace/rl-baselines3-zoo-master$ python enjoy.py --algo tqc --env parking-v0 --folder logs/tqc/parking-v0_1/
Loading latest experiment, id=0
Loading logs/tqc/parking-v0_1/tqc/parking-v0.zip
Traceback (most recent call last):
  File "enjoy.py", line 4, in <module>
    enjoy()
  File "/home/zhuqinghua/workspace/rl-baselines3-zoo-master/rl_zoo3/enjoy.py", line 192, in enjoy
    model = ALGOS[algo].load(model_path, custom_objects=custom_objects, device=args.device, **kwargs)
  File "/home/zhuqinghua/miniconda3/lib/python3.8/site-packages/stable_baselines3/common/base_class.py", line 737, in load
    model._setup_model()
  File "/home/zhuqinghua/miniconda3/lib/python3.8/site-packages/sb3_contrib/tqc/tqc.py", line 153, in _setup_model
    super()._setup_model()
  File "/home/zhuqinghua/miniconda3/lib/python3.8/site-packages/stable_baselines3/common/off_policy_algorithm.py", line 187, in _setup_model
    assert self.env is not None, "You must pass an environment when using `HerReplayBuffer`"
AssertionError: You must pass an environment when using `HerReplayBuffer`

Am I wrong in any step? How can I fix the error?

Checklist

moneypi commented 5 months ago

python enjoy.py --algo tqc --env parking-v0 -f logs/ --exp-id 0