update iluvatar mobilenetv2 config (#356)

Co-authored-by: sen.li <sen.li@iluvatar.com>
FlagOpen · Dec 7, 2023 · 1edbda2 · 1edbda2
1 parent 5028b58
commit 1edbda2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/training/iluvatar/mobilenetv2-pytorch/README.md b/training/iluvatar/mobilenetv2-pytorch/README.md
@@ -40,7 +40,8 @@
 
 | 配置                  | precision | fix_hp         | e2e_time | p_whole | p_train | p_core | acc    | mem         |
 | --------------------- | --------- | -------------- | -------- | ------- | ------- | ------ | ------ | ----------- |
-| BI-V100单机8卡（1x8） | fp32      | bs=256,lr=0.72 | 103759   | 3520    | 3604    | 3651   | 68.61% | 21.6 / 32.0 |
+| BI-V100单机8卡（1x8）  | fp32      | /              | 174534    | 1857    | 1876    | 1885   | 68.52% | 3.6/32.0  |
+| BI-V100单机8卡（1x8） | fp32      | bs=256,lr=0.72 | 87559   | 4390    | 4543    | 4625   | 61.92% | 21.6 / 32.0 |
 | BI-V100单机8卡（1x1） | fp32      | bs=256,lr=0.72 | /        | 624     | 632     | 633    | /      | 21.4 / 32.0 |
 | BI-V100单机8卡（2x8） | fp32      | bs=256,lr=0.72 | /        | 6835    | 7058    | 7219   | /      | 22.2 / 32.0 |
 
diff --git a/training/iluvatar/mobilenetv2-pytorch/config/config_BI-V100x1x8.py b/training/iluvatar/mobilenetv2-pytorch/config/config_BI-V100x1x8.py
@@ -1,5 +1,5 @@
 from config_common import *
 
-train_batch_size = 256
-eval_batch_size = 256
+train_batch_size = 32
+eval_batch_size = 32