Upload README.md with huggingface_hub

README.md

@@ -0,0 +1,87 @@
+---
+language: en
+license: mit
+library_name: timm
+tags:
+- image-classification
+- densenet121
+- cifar10
+datasets: cifar10
+metrics:
+- accuracy
+model-index:
+- name: densenet121_cifar10
+  results:
+  - task:
+      type: image-classification
+    dataset:
+      name: CIFAR-10
+      type: cifar10
+    metrics:
+    - type: accuracy
+      value: 0.9391
+---
+
+# Model Card for Model ID
+
+This model is a small densenet121 trained on cifar10.
+
+- **Test Accuracy:** 0.9391
+- **License:** MIT
+
+## How to Get Started with the Model
+
+Use the code below to get started with the model.
+
+```python
+import detectors
+import timm
+
+model = timm.create_model("densenet121_cifar10", pretrained=True)
+```
+
+## Training Data
+
+Training data is cifar10.
+
+## Training Hyperparameters
+
+
+- **config**: `scripts/train_configs/cifar10.json`
+
+- **model**: `densenet121_cifar10`
+
+- **dataset**: `cifar10`
+
+- **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, 'nesterov': 'True'}`
+
+- **scheduler**: `ReduceLROnPlateau`
+
+- **scheduler_kwargs**: `{'factor': 0.1, 'patience': 3, 'threshold': 0.001, 'mode': 'max'}`
+
+- **debug**: `False`
+
+
+## Testing Data
+
+Testing data is cifar10.
+
+---
+
+This model card was created by Eduardo Dadalto.