fix: bump epochs batch size to larger ones, update READMEs (#155)

* fix: bump epochs batch size to larger ones

* fix: bump each epoch length to 1000

* fix: install tensorboard

* chore: up readmes

scripts/requirements.txt

@@ -7,3 +7,4 @@ image_dataset_viz
 pytest
 transformers
 datasets
+tensorboard

scripts/run_tests.sh

@@ -18,7 +18,13 @@ run_simple() {
   for dir in $(find ./dist-tests/*-simple -type d)
   do
     cd $dir
-    python main.py --data_path ~/data
+    python main.py --data_path ~/data \
+      --train_batch_size 4 \
+      --eval_batch_size 4 \
+      --num_workers 2 \
+      --max_epochs 2 \
+      --train_epoch_length 4 \
+      --eval_epoch_length 4
     cd $CWD
   done
 }
@@ -28,7 +34,13 @@ run_all() {
   do
     cd $dir
     pytest -vra --color=yes --tb=short test_*.py
-    python main.py --data_path ~/data
+    python main.py --data_path ~/data \
+      --train_batch_size 4 \
+      --eval_batch_size 4 \
+      --num_workers 2 \
+      --max_epochs 2 \
+      --train_epoch_length 4 \
+      --eval_epoch_length 4
     cd $CWD
   done
 }
@@ -39,7 +51,13 @@ run_launch() {
     cd $dir
     python -m torch.distributed.launch \
       --nproc_per_node 2 --use_env \
-      main.py --backend gloo --data_path ~/data
+      main.py --backend gloo --data_path ~/data \
+      --train_batch_size 4 \
+      --eval_batch_size 4 \
+      --num_workers 2 \
+      --max_epochs 2 \
+      --train_epoch_length 4 \
+      --eval_epoch_length 4
     cd $CWD
   done
 }
@@ -49,7 +67,13 @@ run_spawn() {
   do
     cd $dir
     python main.py --data_path ~/data \
-      --nproc_per_node 2 --backend gloo
+      --nproc_per_node 2 --backend gloo \
+      --train_batch_size 4 \
+      --eval_batch_size 4 \
+      --num_workers 2 \
+      --max_epochs 2 \
+      --train_epoch_length 4 \
+      --eval_epoch_length 4
     cd $CWD
   done
 }

src/templates/template-common/config.yaml

@@ -1,11 +1,11 @@
 seed: 666
 data_path: ./
-train_batch_size: 4
-eval_batch_size: 4
-num_workers: 2
-max_epochs: 2
-train_epoch_length: 4
-eval_epoch_length: 4
+train_batch_size: 32
+eval_batch_size: 32
+num_workers: 4
+max_epochs: 20
+train_epoch_length: 1000
+eval_epoch_length: 1000
 use_amp: false
 debug: false
 

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
 

src/templates/template-text-classification/config.yaml

@@ -1,11 +1,11 @@
 seed: 666
 data_path: ./
-train_batch_size: 4
-eval_batch_size: 4
-num_workers: 2
-max_epochs: 2
-train_epoch_length: 4
-eval_epoch_length: 4
+train_batch_size: 32
+eval_batch_size: 32
+num_workers: 4
+max_epochs: 20
+train_epoch_length: 1000
+eval_epoch_length: 1000
 use_amp: false
 debug: false
 model: bert-base-uncased

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
 

src/templates/template-vision-classification/config.yaml

@@ -1,11 +1,11 @@
 seed: 666
 data_path: ./
-train_batch_size: 4
-eval_batch_size: 4
-num_workers: 2
-max_epochs: 2
-train_epoch_length: 4
-eval_epoch_length: 4
+train_batch_size: 32
+eval_batch_size: 32
+num_workers: 4
+max_epochs: 20
+train_epoch_length: 1000
+eval_epoch_length: 1000
 lr: 0.0001
 use_amp: false
 debug: false

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
 

src/templates/template-vision-dcgan/config.yaml

@@ -1,11 +1,11 @@
 seed: 666
 data_path: ./
-train_batch_size: 4
-eval_batch_size: 4
-num_workers: 2
-max_epochs: 2
-train_epoch_length: 4
-eval_epoch_length: 4
+train_batch_size: 32
+eval_batch_size: 32
+num_workers: 4
+max_epochs: 20
+train_epoch_length: 1000
+eval_epoch_length: 1000
 lr: 0.0001
 use_amp: false
 debug: false

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
 

src/templates/template-vision-segmentation/config.yaml

@@ -1,11 +1,11 @@
 seed: 666
 data_path: ./
-train_batch_size: 4
-eval_batch_size: 4
-num_workers: 2
-max_epochs: 2
-train_epoch_length: 4
-eval_epoch_length: 4
+train_batch_size: 32
+eval_batch_size: 32
+num_workers: 4
+max_epochs: 20
+train_epoch_length: 1000
+eval_epoch_length: 1000
 lr: 0.007
 use_amp: false
 debug: false