Empowerment for surprise minimization in gridworld and overcooked environments.
To install required packages, use conda env create
To train the learned contrastive empowerment agent, activate the virtual environment using conda activate empgrid, and run train.py with the desired arguments:
python train.py --num_boxes 0 --seed 0