SmilingWolf
commited on
Commit
•
9e6d6a7
1
Parent(s):
af7f33f
Amend the JAX model config file: add image size.
Browse files- README.md +4 -0
- 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 |
+
}
|