# View-based Exploration
Task-driven exploration training code of ViewX.
This codebase is built on [RIDE](https://github.com/facebookresearch/impact-driven-exploration) and [NovelD](https://github.com/tianjunz/NovelD).
The licenses in the code are directly copied from the original codebase. 

## Installation

```
# Install Instructions
conda create -n viewx python=3.7
conda activate viewx 
pip install -r requirements.txt
```

## Train ViewX on MiniGrid
```
. ./example_script.sh
```

## Train the baselines on MiniGrid
```
OMP_NUM_THREADS=1 python main.py --model bebold --env MiniGrid-ObstructedMaze-Full-v0 --total_frames 50000000 --intrinsic_reward_coef 0.01 --count_reward_coef 0.01
```
NOTE: bebold is the same as NovelD
