From 2159e890657c46ce205017dae835218336d07bea Mon Sep 17 00:00:00 2001 From: Vaibhav Balloli Date: Mon, 12 Oct 2020 02:49:12 +0200 Subject: [PATCH] Update README.md --- README.md | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 48d0899..5e8f947 100644 --- a/README.md +++ b/README.md @@ -23,7 +23,13 @@ img = jax.random.uniform(rng, (1,256,256,3)) output = model(img) ``` -Note: This repository is still in initial stages. Feel free to [Contact me](https://vballoli.github.io) or raise issues/PR for suggestions, improvements or bugs. +`examples` directory contains code to train ViT on CIFAR datasets. +# Docs and references + +Documentation for all the modules can be viewed [here](https://vit-flax.readthedocs.io/en/main/). + +# Note +This repository is still in initial stages. Feel free to [Contact me](https://vballoli.github.io) or raise issues/PR for suggestions, improvements or bugs. # Help needed