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.2570, F1 = 0.4282`
+`v2.0: P=R: threshold = 0.2682, F1 = 0.4419`  
+`v1.0: P=R: threshold = 0.2570, F1 = 0.4282`
 
 ## 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:c5e639920c05e5cc100dc0dd7e5002c9eb39367a227aa3dbd1f942068c76aeee
+oid sha256:4294aaa66dad2bb01e4552487c7f1402e9a7409744c1fc1f90e70b5ac8f3b5cf
 size 394808412

model.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:02f30d4de9bada756981a11464d13aa206f5e2d4ff6da384511beb812d58b2ca
+oid sha256:1b8a7abf13d9b8368267df47501d523789c4aeae66b2296ad98483239dfa32eb
 size 394990732

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7f99a11a43a459eca201e7e4dbc0d02d980c9e0f5d8250ed4d07f69927e3b30a
+oid sha256:77ad172eb1f7a697dc629bef7da7d50402e93ef0b1c86d7f69a865d31be64022
 size 394830604

sw_jax_cv_config.json

@@ -10,4 +10,4 @@
         "use_conv_bias": true,
         "layer_norm_eps": 1e-06
     }
-}
+}