ngopaul/pytorch-curiosity-norewardrl

Curiosity-driven Exploration by Self-supervised Prediction

1. Setup

Requirements

---

• python3.6
• gym
• OpenCV Python
• PyTorch
• tensorboardX
• atari roms

2. How to Train

Modify the parameters in config.conf.

python train.py

In a separate terminal:

tensorboard --logdir runs

3. How to Eval

python eval.py

References

[1] Actor-Critic Algorithms
[2] Efficient Parallel Methods for Deep Reinforcement Learning
[3] Curiosity-driven Exploration by Self-supervised Prediction
[4] Proximal Policy Optimization Algorithms
[5] Large-Scale Study of Curiosity-Driven Learning