sw_jax_cv_config.json

{
    "image_size": 448,
    "model_name": "vit_large",
    "model_args": {
        "patch_size": 14,
        "num_classes": 10861,
        "num_layers": 24,
        "embed_dim": 1024,
        "mlp_dim": 4096,
        "num_heads": 16,
        "drop_path_rate": 0.1,
        "norm_layer": "reparam_layernorm",
        "layer_norm_eps": 1e-05
    }
}