Amend the JAX model config file: add image size.

README.md

@@ -20,6 +20,10 @@ Tags with less than 600 images were filtered out.
 `P=R: threshold = 0.2547, F1 = 0.4278`
 
 ## What's new
+Model v1.1/Dataset v3:  
+Amended the JAX model config file: add image size.
+No change to the trained weights.
+
 Model v1.0/Dataset v3:  
 More training images, more and up-to-date tags (up to 2024-02-28).  
 Now `timm` compatible! Load it up and give it a spin using the canonical one-liner!  

sw_jax_cv_config.json

@@ -1,4 +1,5 @@
 {
+    "image_size": 448,
     "model_name": "vit_base",
     "model_args": {
         "patch_size": 16,
@@ -10,4 +11,4 @@
         "drop_path_rate": 0.1,
         "layer_norm_eps": 1e-05
     }
-}
+}