
# NeuroTrails: Training with Dynamic Sparse Heads as the Key to Effective Ensembling


## Install

Tested on Linux.

```
conda create -n natari python=3.10
conda activate natari
pip install -r requirements-atari.txt
```


## Run

To train various models, use commands like these:

```
# Single DQN
python dqn_atari.py --no-track

# Full Ensemble
python dqn_atari.py --no-track --num_ensemble 3

# TreeNet
python dqn_atari.py --no-track --num_ensemble 3 --blocks_in_head 4

# NeuroTrails
python dqn_atari.py --no-track --num_ensemble 3 --blocks_in_head 4 --density 0.2
```


