bwang0911 commited on
Commit
98a5f64
·
verified ·
1 Parent(s): c6d2f3b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +16 -3
README.md CHANGED
@@ -1,3 +1,16 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ ---
4
+
5
+ ## Cat Embeddings
6
+
7
+ A set of embedding model trained for study embedding quality vs model architecture (width/depth) given a size constraint (12M params).
8
+
9
+ - **cat-emb-2-128**: 2 layers/hidden size 128/4.4m
10
+ - **cat-emb-4-128**: 4 layers/H 128/4.8m
11
+ - **cat-emb-6-128**: 6 layers/H 128/5.2m
12
+ - **cat-emb-8-128**: 8 layers/H 128/5.6m
13
+ - **cat-emb-10-128**: 10 layers/H 128/6.0m
14
+ - **cat-emb-12-128**: 12 layers/H 128/6.4m
15
+ - **cat-emb-2-256**: 2 layers/H 256/9.7m
16
+ - **cat-emb-4-256**: 4 layers/H 256/11.3m