settings.json

{
    "backbone": "vit",
    "params": {
        "model_size": "base",
        "patch_size": 32,
        "image_size": 448,
        "embedding_size": 128,
        "add_pooling_layer": false,
        "mean_pooling": false,
        "dropout_dense": 0.1,
        "device": "cuda",
        "force_download": true
    }
}