This model is a small densenet121 trained on svhn.
Use the code below to get started with the model.
1import detectors
2import timm
3
4model = timm.create_model("densenet121_svhn", pretrained=True)
Training data is svhn.
-
config: scripts/train_configs/svhn.json
-
model: densenet121_svhn
-
dataset: svhn
-
batch_size: 128
-
epochs: 300
-
validation_frequency: 5
-
seed: 1
-
criterion: CrossEntropyLoss
-
criterion_kwargs: {}
-
optimizer: SGD
-
lr: 0.01
-
optimizer_kwargs: {'momentum': 0.9, 'weight_decay': 0.0005}
-
scheduler: MultiStepLR
-
scheduler_kwargs: {'gamma': 0.1, 'milestones': [75, 100, 150, 225]}
-
debug: False
Testing data is svhn.
This model card was created by Eduardo Dadalto.
