SmilingWolf commited on
Commit
9e6d6a7
1 Parent(s): af7f33f

Amend the JAX model config file: add image size.

Browse files
Files changed (2) hide show
  1. README.md +4 -0
  2. sw_jax_cv_config.json +2 -1
README.md CHANGED
@@ -20,6 +20,10 @@ Tags with less than 600 images were filtered out.
20
  `P=R: threshold = 0.2521, F1 = 0.4411`
21
 
22
  ## What's new
 
 
 
 
23
  Model v1.0/Dataset v3:
24
  More training images, more and up-to-date tags (up to 2024-02-28).
25
  Now `timm` compatible! Load it up and give it a spin using the canonical one-liner!
 
20
  `P=R: threshold = 0.2521, F1 = 0.4411`
21
 
22
  ## What's new
23
+ Model v1.1/Dataset v3:
24
+ Amended the JAX model config file: add image size.
25
+ No change to the trained weights.
26
+
27
  Model v1.0/Dataset v3:
28
  More training images, more and up-to-date tags (up to 2024-02-28).
29
  Now `timm` compatible! Load it up and give it a spin using the canonical one-liner!
sw_jax_cv_config.json CHANGED
@@ -1,4 +1,5 @@
1
  {
 
2
  "model_name": "swinv2_base",
3
  "model_args": {
4
  "image_size": 448,
@@ -15,4 +16,4 @@
15
  "patch_norm": true,
16
  "layer_norm_eps": 1e-05
17
  }
18
- }
 
1
  {
2
+ "image_size": 448,
3
  "model_name": "swinv2_base",
4
  "model_args": {
5
  "image_size": 448,
 
16
  "patch_norm": true,
17
  "layer_norm_eps": 1e-05
18
  }
19
+ }