Update README.md

README.md

@@ -9,7 +9,22 @@ tags:
 
 ## Model description
 
-More information needed
+### Consistency training with supervision
+
+In this example, we have trained an image classification model enforcing a sense of consistency inside it by doing the following:
+
+- Train a standard image classification model.
+- Train an equal or larger model on a noisy version of the dataset (augmented using RandAugment).
+- To do this, we will first obtain predictions of the previous model on the clean images of the dataset.
+- We will then use these predictions and train the second model to match these predictions on the noisy variant of the same images. This is identical to the workflow of Knowledge Distillation but since the student model is equal or larger in size this process is also referred to as Self-Training.
+
+This overall training workflow finds its roots in works like FixMatch, Unsupervised Data Augmentation for Consistency Training, and Noisy Student Training. Since this training process encourages a model yield consistent predictions for clean as well as noisy images, it's often referred to as consistency training or training with consistency regularization. Although the example focuses on using consistency training to enhance the robustness of models to common corruptions this example can also serve a template for performing weakly supervised learning.
+
+Full Credits to <a href = "https://twitter.com/RisingSayak" target='_blank'> Sayak Paul </a> for this work.
+
+This repo contains only the <b> Teacher Model </b> of this training example.
+
+<b>Student Model </b>Repo can be find at this <a href = "" target='_blank'> Link </a>.
 
 ## Intended uses & limitations
 
@@ -17,7 +32,8 @@ More information needed
 
 ## Training and evaluation data
 
-More information needed
+Trained and evaluated on [CIFAR-10](https://keras.io/api/datasets/cifar10/) dataset.
+
 
 ## Training procedure