Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
zwxandy authored Jun 30, 2023
1 parent 5a8f88e commit 1ffae8d
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -63,6 +63,7 @@ You can evaluate MPCViT+ models on CIFAR-10/100 with the following command:
```shell
python inference.py --config configs/datasets/cifar10.yml --model vit_7_4_32 /path/to/cifar10 --model-checkpoint mpcvit_checkpoints/cifar-10/mpcvit_plus_cifar10.pth.tar --linear-gelu
```
Note that some data points need post-added ReLU after GeLU linearization.

## Citation
```bibtex
Expand Down

0 comments on commit 1ffae8d

Please sign in to comment.