This repo contains code for the NeurIPS submission "A General Theory on Correct, Incorrect, and Extrinsic Equivariance"
- Install PyTorch (recommended 1.10.2)
pip install -r requirements.txt
cd spiral
python main.py --model=dssz // invariant network
python main.py --model=mlp // unconstained network
cd square
python main.py
cd regression
python main.py
cd mnist
python main.py --model=d4 // D4 network
python main.py --model=cnn // CNN networ
cd print_digit
python main.py --model=d4 // D4 network
python main.py --model=cnn // CNN networ
cd robotic
See the README.md in the robotic subfolder.