This repo provides the official implementation of the paper. The code is based on original spr implementation.
To install the requirements, follow these steps:
# PyTorch
pip install pytorch torchvision
# Install requirements
pip install -r requirements.txt- To run Eqr with group equivariant loss
python -m scripts.run --game krull --momentum-tau 1. --seed 0 --group-type 'En' --num-blocks 12 --projection-type 'q_shared' --second-projection-type 'mlp' --acteqv-loss-weight 1. --groupeqv-loss-weight 1. --reward-loss-weight 1.- To run Eqr with just action equivariant loss
python -m scripts.run --game krull --momentum-tau 1. --seed 0 --group-type 'En' --num-blocks 12 --projection-type 'q_shared' --second-projection-type 'mlp' --acteqv-loss-weight 1. --groupeqv-loss-weight 0. --reward-loss-weight 1.