Skip to content

Commit

Permalink
Adds prominent link to big_vision codebase.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 580098659
  • Loading branch information
andsteing authored and copybara-github committed Nov 7, 2023
1 parent ac6e056 commit ab070fc
Showing 1 changed file with 9 additions and 0 deletions.
9 changes: 9 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,15 @@ The models were pre-trained on the [ImageNet](http://www.image-net.org/) and
fine-tuning the released models in
[JAX](https://jax.readthedocs.io)/[Flax](http://flax.readthedocs.io).

The models from this codebase were originally trained in
https://github.com/google-research/big_vision/
where you can find more advanced code (e.g. multi-host training), as well as
some of the original training scripts (e.g.
[configs/vit_i21k.py](https://github.com/google-research/big_vision/blob/main/big_vision/configs/vit_i21k.py)
for pre-training a ViT, or
[configs/transfer.py](https://github.com/google-research/big_vision/blob/main/big_vision/configs/transfer.py)
for transfering a model).

Table of contents:

- [Vision Transformer and MLP-Mixer Architectures](#vision-transformer-and-mlp-mixer-architectures)
Expand Down

0 comments on commit ab070fc

Please sign in to comment.