This model is a small resnet34 trained on cifar10.
Use the code below to get started with the model.
1import detectors
2import timm
3
4model = timm.create_model("resnet34_supcon_cifar10", pretrained=True)
Training data is cifar10.
-
config: None
-
model: resnet34_supcon_cifar10
-
batch_size: 512
-
epochs: 501
-
lr: 0.5
-
warmup_epochs: 10
-
validation_frequency: 50
-
output_features_dim: 128
-
seed: 1
-
debug: False
-
dataset: cifar10
-
training_mode: supcon
Testing data is cifar10.
This model card was created by Eduardo Dadalto.
