Update model files

README.md

@@ -17,11 +17,16 @@ Images with less than 10 general tags were filtered out.
 Tags with less than 600 images were filtered out.
 
 ## Validation results
-`P=R: threshold = 0.2547, F1 = 0.4278`
+`v2.0: P=R: threshold = 0.2614, F1 = 0.4402`  
+`v1.0: P=R: threshold = 0.2547, F1 = 0.4278`
 
 ## What's new
+Model v2.0/Dataset v3:  
+Trained for a few more epochs.  
+Used tag frequency-based loss scaling to combat class imbalance.
+
 Model v1.1/Dataset v3:  
-Amended the JAX model config file: add image size.
+Amended the JAX model config file: add image size.  
 No change to the trained weights.
 
 Model v1.0/Dataset v3:  
@@ -31,6 +36,14 @@ ONNX model is compatible with code developed for the v2 series of models.
 The batch dimension of the ONNX model is not fixed to 1 anymore. Now you can go crazy with batch inference.  
 Switched to Macro-F1 to measure model performance since it gives me a better gauge of overall training progress.
 
+# Runtime deps
+ONNX model requires `onnxruntime >= 1.17.0`
+
+# Inference code examples
+For timm: https://github.com/neggles/wdv3-timm  
+For ONNX: https://huggingface.co/spaces/SmilingWolf/wd-tagger
+For JAX: https://github.com/SmilingWolf/wdv3-jax
+
 ## Final words
 Subject to change and updates.
 Downstream users are encouraged to use tagged releases rather than relying on the head of the repo.

model.msgpack

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b06edd4477d3a5407afbc41a2e0d36773bc00e549eda5d19ad0395a0d23618bd
+oid sha256:a4b68183d5db1191ed4dbe0bfa816483454e55612b4e3a0c76e06229734c4ed6
 size 378408630

model.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0c21152f1febcd04867c4dbc7bddf781db701f7c0e14fa3b787656477a16514c
+oid sha256:35f23693620b668f4d53fd3c62bf65e40af739bc52c7eb0fbc49258b58d065b6
 size 378536310

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:021ace976c1e9f944ea51d591c3f6a5722125ae8bd21c51caec3fd1275587ed5
+oid sha256:ee78db0cd9f1c7e01578cc89661a10c73675d60c30e235e505c391f57ea4d4e5
 size 378417260

sw_jax_cv_config.json

@@ -11,4 +11,4 @@
         "drop_path_rate": 0.1,
         "layer_norm_eps": 1e-05
     }
-}
+}