chore: up readmes

jeffydc · jeffydc · commit 144870165b8e · 2021-06-03T15:28:28.000+06:30
diff --git a/src/templates/template-text-classification/README.md b/src/templates/template-text-classification/README.md
@@ -1,4 +1,8 @@
-# Template by Code-Generator
+[![Code-Generator](https://badgen.net/badge/Template%20by/Code-Generator/ee4c2c?labelColor=eaa700)](https://github.com/pytorch-ignite/code-generator)
+
+# Text Classification Template
+
+This is the text classification template by Code-Generator using `bert-base-uncased` model from HuggingFace Transformers and `imdb` dataset from HuggingFace datasets and training is powered by PyTorch and PyTorch-Ignite.
 
 ## Getting Started
 
diff --git a/src/templates/template-vision-classification/README.md b/src/templates/template-vision-classification/README.md
@@ -1,4 +1,8 @@
-# Template by Code-Generator
+[![Code-Generator](https://badgen.net/badge/Template%20by/Code-Generator/ee4c2c?labelColor=eaa700)](https://github.com/pytorch-ignite/code-generator)
+
+# Image Classification Template by Code-Generator
+
+This is the image classification template by Code-Generator using `resnet18` model and `cifar10` dataset from TorchVision and training is powered by PyTorch and PyTorch-Ignite.
 
 ## Getting Started
 
diff --git a/src/templates/template-vision-dcgan/README.md b/src/templates/template-vision-dcgan/README.md
@@ -1,4 +1,8 @@
-# Template by Code-Generator
+[![Code-Generator](https://badgen.net/badge/Template%20by/Code-Generator/ee4c2c?labelColor=eaa700)](https://github.com/pytorch-ignite/code-generator)
+
+# DCGAN Template
+
+This is the dcgan template by Code-Generator using `dcgan` model and `cifar10` dataset from TorchVision and training is powered by PyTorch and PyTorch-Ignite.
 
 ## Getting Started
 
diff --git a/src/templates/template-vision-segmentation/README.md b/src/templates/template-vision-segmentation/README.md
@@ -1,4 +1,8 @@
-# Template by Code-Generator
+[![Code-Generator](https://badgen.net/badge/Template%20by/Code-Generator/ee4c2c?labelColor=eaa700)](https://github.com/pytorch-ignite/code-generator)
+
+# Segmentation Template
+
+This is the segmentation template by Code-Generator using `deeplabv3_resnet101` and `cifar10` dataset from TorchVision and training is powered by PyTorch and PyTorch-Ignite.
 
 ## Getting Started
 
