SmilingWolf commited on
Commit
76f1f3e
1 Parent(s): b498930

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.2570, F1 = 0.4282`
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.2570, F1 = 0.4282`
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": "convnext_base",
3
  "model_args": {
4
  "image_size": 448,
@@ -9,4 +10,4 @@
9
  "use_conv_bias": true,
10
  "layer_norm_eps": 1e-06
11
  }
12
- }
 
1
  {
2
+ "image_size": 448,
3
  "model_name": "convnext_base",
4
  "model_args": {
5
  "image_size": 448,
 
10
  "use_conv_bias": true,
11
  "layer_norm_eps": 1e-06
12
  }
13
+ }