search

Shuijing725 / CrowdNav_DSRNN

[ICRA 2021] Decentralized Structural-RNN for Robot Crowd Navigation with Deep Reinforcement Learning
https://sites.google.com/illinois.edu/crowdnav-dsrnn/home
MIT License
97 stars 23 forks source link

train #4

Closed cir-lab closed 3 years ago

cir-lab commented 3 years ago

Hello, I am very sorry to disturb you. I noticed you said the environment configurations file is crowd_nav/configs/env.config，but I can‘t find this file. image

Shuijing725 commented 3 years ago

Sorry, it should be crowd_nav/configs/config.py. The readme is updated now.

cir-lab commented 3 years ago

I am very sorry to disturb you again. When I run the program python test.py, I encountered problem “ 'GRU' object has no attribute 'proj_size' ” image

Shuijing725 commented 3 years ago

It will be easier for me to diagnose if you could show the full error message.

cir-lab commented 3 years ago

image The full error message is shown in the figure. I would be very grateful if you can help me find the problem.

Shuijing725 commented 3 years ago

It looks like the error is caused by PyTorch data parallelism, see https://discuss.pytorch.org/t/issues-using-data-parallelism-dataparallel-object-has-no-attribute-xxxxxxxx/94102. If you only have one gpu, can you try to replace nn.DataParallel(actor_critic).to(device) with actor_critic.to(device)? If you need to train it on multiple gpus, can you try to replace it with nn.DataParallel(actor_critic).module.to(device)?

mincheulkim commented 3 years ago

It might be caused by using different version of pyTorch and torchvision. What about using below version of software?

torch==1.7.1 torchvision==0.1.8

Shuijing725 commented 3 years ago

Good catch @mincheulkim ! Previously I used PyTorch 1.5.0. If you want to load the example model, this repo supports up to PyTorch 1.7.1. The requirements.txt is updated.