RND-Pytoch

Random Network Distillation(RND) algo in pytorch

This is a PyTorch implementation of Exploration by Random Network Distillation paper.

Requirements

• python3.6
• PyTorch
• gym
• numpy
• PIL
• argparse
• tensorboardX

Usage:

Prepare

CUDA_VISIBLE_DEVICES=0

Training

python train.py

Enjoy

python enjoy.py

TensorboardX Graph

tensorboard --logdir runs

Open in browser http://localhost:6006