toshikwa / discor.pytorch Goto Github PK

Hi,

I met a problem when I run the test.py
python test.py --cuda --render --env_id Walker2d-v2 --log_dir ~/discor.pytorch/logs/Walker2d-v2/discor-seed0-20210608-2206 /home/jiayao/anaconda3/envs/discor/lib/python3.8/site-packages/gym/logger.py:30: UserWarning: WARN: Box bound precision lowered by casting to float32 warnings.warn(colorize('%s: %s'%('WARN', msg % args), 'yellow')) Traceback (most recent call last): File "test.py", line 83, in <module> run(args) File "test.py", line 51, in run algo.load_models(os.path.join(args.log_dir, 'model', 'best')) File "/home/jiayao/discor.pytorch/discor/algorithm/eval.py", line 23, in load_models self._policy_net.load(os.path.join(save_dir, 'policy_net.pth')) File "/home/jiayao/discor.pytorch/discor/network.py", line 39, in load self.load_state_dict(torch.load(path)) File "/home/jiayao/anaconda3/envs/discor/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1223, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for GaussianPolicy: Missing key(s) in state_dict: "net.6.weight", "net.6.bias". size mismatch for net.4.weight: copying a param with shape torch.Size([12, 256]) from checkpoint, the shape in current model is torch.Size([256, 256]). size mismatch for net.4.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([256]).

Could you help me solve the problem? I would appreciate it.
Cheers!

Hi,

When I run python train.py --cuda --env_id Walker2d-v2 --config config/mujoco.yaml --algo discor

I met the following errors:
/home/jiayao/anaconda3/envs/discor/lib/python3.8/site-packages/gym/logger.py:30: UserWarning: WARN: Box bound precision lowered by casting to float32 warnings.warn(colorize('%s: %s'%('WARN', msg % args), 'yellow')) Traceback (most recent call last): File "train.py", line 7, in <module> from discor.env import make_env File "/home/jiayao/discor.pytorch/discor/env.py", line 3, in <module> from metaworld.envs.mujoco.env_dict import ALL_ENVIRONMENTS ImportError: cannot import name 'ALL_ENVIRONMENTS' from 'metaworld.envs.mujoco.env_dict' (/home/jiayao/anaconda3/envs/discor/lib/python3.8/site-packages/metaworld/envs/mujoco/env_dict.py)

So I come to check env_dict.py and indeed there's no ALL_ENVIRONMENTS but ALL_V1_ENVIRONMENTS and ALL_V2_ENVIRONEMTS.

Could you help me solve this probelm. I would appreciate that.
Cheers!