Missing folders

[SliencerX]

Hi,
this is a great work and thanks for sharing the code! In script_new.sh file, it mentions that some RL implementations are in folders atari and radial_rl-cleanrl/DQN, but they are not in the repository.

[ghost]

I've uploaded the two folders ("atari" and "radial_rl-cleanrl") that contain the code for training. They are basically adapted from prior repos as described in README. Let me know if you have further questions!

Thanks,
Fan

[SliencerX]

Thanks for much for uploading those folders. When I tried to train the model, I ran into an error message of "CUDA error: an illegal memory access was encoutered" (randomly appeared during the training). Could this be caused by different pytorch and cuda version? It will be great if you could share the environment settings.

Thanks,
Xiaolin

[ghost]

CUDA version: 11.2
PyTorch version: 1.8.0+cu111

We recommend you to also check out the following repo for the code of model training, which our code is adapted upon.
https://github.com/chenhongge/SA_DQN
https://github.com/tuomaso/radial_rl
https://github.com/vwxyzjn/cleanrl