| Experiment | Model | Dataset | Num Samples | Problem Type | Backend |
|---|---|---|---|---|---|
| 1 | ResNet50 (CNN) | CIFAR100 | 50,000 train, 10,000 test | Image Classification | PyTorch |
| 2 | ResNet50 | Food101 | 75,750 train, 25,250 test | Image Classification | PyTorch |
| 3 | DistilBERT (Transformer) | IMDB | 25,000 train, 25,000 test | Text Classification | PyTorch |
| 4 | ResNet50 | CIFAR100 | 50,000 train, 10,000 test | Image Classification | TensorFlow |
| 5 | ResNet50 | Food101 | 75,750 train, 25,250 test | Image Classification | TensorFlow |
| 6 | SmallTransformer | IMDB | 25,000 train, 25,000 test | Text Classification | TensorFlow |
| 7 | Llama 2 7B Q4_0.gguf | N/A | 100 (generate 100 answers) | Text Generation | llama.cpp |
| 8 (bonus) | Geekbench ML | Multiple | Multiple | Multiple | Core ML |
Created
January 9, 2024 00:42
-
-
Save mrdbourke/37f243dbb675d05835928dd8bd88a1d8 to your computer and use it in GitHub Desktop.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment