benleader commited on
Commit
b7d7dcc
·
1 Parent(s): f50d964

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -30,14 +30,14 @@ We use the Baichuan2-7B-Base and Baichuan2-13B-Base model for measurement, but t
30
  | Version | Batch Size 1 | Batch Size 8 | Batch Size 16 | Batch Size 32 | Batch Size 64 |
31
  | --- | --- | --- | --- | --- | --- |
32
  | Torch 2.0.1 | 41.2 | 323.2 | 640.0 | 1256.8 | 2231.0 |
33
- | lyraXVERSE MEMOPT | 125.9 | 948.1 | 1749.3 | 2974.0 | 4370.1 |
34
 
35
  ### Baichuan2-13B-Base
36
 
37
  | Version | Batch Size 1 | Batch Size 8 | Batch Size 16 | Batch Size 32 | Batch Size 64 |
38
  | --- | --- | --- | --- | --- | --- |
39
  | Torch 2.0.1 | 40.9 | 307.9 | 555.6 | 1010.4 | 1601.0 |
40
- | lyraXVERSE MEMOPT | 80.0 | 568.2 | 1124.4 | 1942.6 | 2828.0 |
41
 
42
  ## Docker Environment Recommendation
43
 
 
30
  | Version | Batch Size 1 | Batch Size 8 | Batch Size 16 | Batch Size 32 | Batch Size 64 |
31
  | --- | --- | --- | --- | --- | --- |
32
  | Torch 2.0.1 | 41.2 | 323.2 | 640.0 | 1256.8 | 2231.0 |
33
+ | lyraBaichuan | 125.9 | 948.1 | 1749.3 | 2974.0 | 4370.1 |
34
 
35
  ### Baichuan2-13B-Base
36
 
37
  | Version | Batch Size 1 | Batch Size 8 | Batch Size 16 | Batch Size 32 | Batch Size 64 |
38
  | --- | --- | --- | --- | --- | --- |
39
  | Torch 2.0.1 | 40.9 | 307.9 | 555.6 | 1010.4 | 1601.0 |
40
+ | lyraBaichuan | 80.0 | 568.2 | 1124.4 | 1942.6 | 2828.0 |
41
 
42
  ## Docker Environment Recommendation
43