bswift
/

RETfound_eyepacs_DR

Image Classification

Inference Endpoints

Model card Files Files and versions Community

bswift commited on Dec 12, 2023

Commit

067ee07

•

1 Parent(s): 70110c1

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -1,3 +1,3 @@
-{"model_type": "ViT", "architectures": [
     "ViTForImageClassification"
   ], "input_size": 256, "num_classes": 5, "drop_path_rate": 0.2, "layer_decay": 0.65, "weight_decay": 0.05, "base_learning_rate": 0.005, "batch_size": 10, "epochs": 50, "remove_background": true}

+{"model_type": "vit", "architectures": [
     "ViTForImageClassification"
   ], "input_size": 256, "num_classes": 5, "drop_path_rate": 0.2, "layer_decay": 0.65, "weight_decay": 0.05, "base_learning_rate": 0.005, "batch_size": 10, "epochs": 50, "remove_background": true}