Never give up

PyTorch implementation of Never Give Up: Learning Directed Exploration Strategies [Paper]

Only the episodic curiosity with embedding network has been implemented.

Install

Tested with Python 3.7.9

pip install -r requirements.txt

Train

python train.py

Results

Results on 5x5 gym-maze

reward

Credits

R2D2 base is from here by g6ling