Image Classification
timm
Safetensors
image classification
emotion-recognition
PyTorch
Transformers
Safetensors
DeiT
Instructions to use Simon14142/emotion-recognition-4class-deit with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- timm
How to use Simon14142/emotion-recognition-4class-deit with timm:
import timm model = timm.create_model("hf-hub:Simon14142/emotion-recognition-4class-deit", pretrained=True) - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -79,7 +79,7 @@ DeiT-Base is significantly weaker. Macro-F1 reaches 0.60 and Cohen's Kappa reach
|
|
| 79 |
<br><br>
|
| 80 |
|
| 81 |
## <strong>Hardware information</strong> <br>
|
| 82 |
-
Both models are trained on a single NVIDIA Blackwell GB203 GPU with 16 GB of VRAM (GDDR7, non-ECC, 28 Gigabit/s over 256bit-bus = 896
|
| 83 |
<br>
|
| 84 |
|
| 85 |
At batch-size of 64, training takes around 1 hour for DeiT-Base (approx. 7 batches per second) and 1.5 hours for SWIN-Base (approx. 5.2 batches per second). DeiT-Base needs approx. 6.0 GiB of VRAM while SWIN-Base needs approx. 11.4 GiB of VRAM. Training should therefore be possible on a GPU with 12 GiB of VRAM. We still recommend a GPU with at least 16 GiB of VRAM. DRAM usage stays below 10 GiB for both models. Therefore, 16 GiB of DRAM should suffice, but 24 GiB or more are recommended.
|
|
|
|
| 79 |
<br><br>
|
| 80 |
|
| 81 |
## <strong>Hardware information</strong> <br>
|
| 82 |
+
Both models are trained on a single NVIDIA Blackwell GB203 GPU with 16 GB of VRAM (GDDR7, non-ECC, 28 Gigabit/s over 256bit-bus = 896 GB/s (~ 834 Gib/s) transfer speed). The GPU is connected to the system via PCI-Express Gen. 5.0 with 16 lanes. CPU-side is handled by a 12th Gen. (Alder Lake) Intel Core i5-12600K on a motherboard with a Z690 chipset. Our system has access to 64 GB of DDR4-DRAM (non-ECC) running with JEDEC specification for 3.200 MT/s at CL16-20-20-38 at 1.35V. We use a 1TB (~ 928.2 GiB) SAMSUNG 870QVO SSD as mass storage wich is connected via SATA-III (SATA-600).
|
| 83 |
<br>
|
| 84 |
|
| 85 |
At batch-size of 64, training takes around 1 hour for DeiT-Base (approx. 7 batches per second) and 1.5 hours for SWIN-Base (approx. 5.2 batches per second). DeiT-Base needs approx. 6.0 GiB of VRAM while SWIN-Base needs approx. 11.4 GiB of VRAM. Training should therefore be possible on a GPU with 12 GiB of VRAM. We still recommend a GPU with at least 16 GiB of VRAM. DRAM usage stays below 10 GiB for both models. Therefore, 16 GiB of DRAM should suffice, but 24 GiB or more are recommended.
|