edadaltocg's picture
Upload README.md with huggingface_hub
0c27122
metadata
language: en
license: mit
library_name: timm
tags:
  - image-classification
  - densenet121
  - cifar100
datasets: cifar100
metrics:
  - accuracy
model-index:
  - name: densenet121_cifar100
    results:
      - task:
          type: image-classification
        dataset:
          name: CIFAR-100
          type: cifar100
        metrics:
          - type: accuracy
            value: 0.7619

Model Card for Model ID

This model is a small densenet121 trained on cifar100.

  • Test Accuracy: 0.7619
  • License: MIT

How to Get Started with the Model

Use the code below to get started with the model.

import detectors
import timm

model = timm.create_model("densenet121_cifar100", pretrained=True)

Training Data

Training data is cifar100.

Training Hyperparameters

  • config: scripts/train_configs/cifar100.json

  • model: densenet121_cifar100

  • dataset: cifar100

  • batch_size: 128

  • epochs: 300

  • validation_frequency: 5

  • seed: 1

  • criterion: CrossEntropyLoss

  • criterion_kwargs: {}

  • optimizer: SGD

  • lr: 0.1

  • optimizer_kwargs: {'momentum': 0.9, 'weight_decay': 0.0005}

  • scheduler: CosineAnnealingLR

  • scheduler_kwargs: {'T_max': 280}

  • debug: False

Testing Data

Testing data is cifar100.


This model card was created by Eduardo Dadalto.