update reamdes

leondgarse · May 7, 2022 · 1d3207f · 1d3207f
1 parent f8496f5
commit 1d3207f
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/keras_cv_attention_models/cmt/cmt.py b/keras_cv_attention_models/cmt/cmt.py
@@ -13,7 +13,7 @@
 )
 from keras_cv_attention_models.download_and_load import reload_model_weights
 
-PRETRAINED_DICT = {"cmt_tiny": {"imagenet": {160: "72402495cb42314cedd1e2714f2de893", 224: "b07899f4e1a28bbe5f2f6a705e22ac97"}}}
+PRETRAINED_DICT = {"cmt_tiny": {"imagenet": {160: "cb269248643e9d50d8bea051563e20a6", 224: "d0f2f0cf649a7aea48a1a4e3a476606c"}}}
 
 
 def light_mhsa_with_multi_head_relative_position_embedding(

diff --git a/keras_cv_attention_models/davit/README.md b/keras_cv_attention_models/davit/README.md
@@ -15,6 +15,17 @@
   | DaViT_L, 21k  | 196.8M | 103.2G | 384   | 87.5     |          |
   | DaViT_H, 1.5B | 348.9M | 327.3G | 512   | 90.2     |          |
   | DaViT_G, 1.5B | 1.406B | 1.022T | 512   | 90.4     |          |
+
+  **Self tested accuracy**. There may be some detail differences in model output layer or evaluating process.
+  ```sh
+  CUDA_VISIBLE_DEVICES='0' ./eval_script.py -m davit.DaViT_T
+  # >>>> Accuracy top1: 0.82276 top5: 0.96152
+  ```
+  | Model   | Self tested Top1 Acc |
+  | ------- | -------------------- |
+  | DaViT_T | 82.276               |
+  | DaViT_S | 83.810               |
+  | DaViT_B | 84.142               |
 ## Usage
   ```py
   from keras_cv_attention_models import davit