Table 2 GAN training and testing time on Google Colab Pro GPUs
From: Vi-GeN 1.0: GAN-Augmented Vision Transformer Pipeline for Diversified Pneumonia Classification
Metric | Tesla T4 | A100 |
|---|---|---|
GAN training time (min) | 180 min (3 h) | 90 min (1.5 h) |
GAN testing time (ms/img) | 25 ms | 15 ms |
ViT testing time (ms/img) | 12 ms | 7 ms |