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.2521, F1 = 0.4411`
+`v2.0: P=R: threshold = 0.2653, F1 = 0.4541`  
+`v1.0: P=R: threshold = 0.2521, F1 = 0.4411`
 
 ## 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:bcdeb6cf20b4674c02bb2761109e2ed40a850d72c24de2ee093de143b83fb66d
+oid sha256:a26ae4edbe45fa155ede34257153642b3bf476044a13e0ae6a0c4c131198f444
 size 413777297

model.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:622d33fc180ed3ecbd886a5318059ea03b5a6df93c170662725104c339cb8b9c
+oid sha256:e6774bff34d43bd49f75a47db4ef217dce701c9847b546523eb85ff6dbba1db1
 size 467460978

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:95a626e5dd214c0c0d36894c62296d2c74c9fec8399b4ce6664dfd239bd40816
+oid sha256:39fe04000d5907a0a7616df54af744d43fbd70b5fba7abc3135d3675734c03ef
 size 392149220

sw_jax_cv_config.json

@@ -16,4 +16,4 @@
         "patch_norm": true,
         "layer_norm_eps": 1e-05
     }
-}
+}