[add]上传训练benchmark by z00560161

2020-10-19 20:22:23 +08:00
parent 22b83024f5
commit 82522e2f61
1225 changed files with 345421 additions and 0 deletions
@@ -0,0 +1,18 @@
+pytorch_config:
+    # 基本参数
+    data_url: /home/imagenet/
+    #跑1p时batch_size为1536, 2p为2048，4p为4096，8p时为8196
+    batch_size: 1536
+    epoches: 10
+    warm_up_epochs: 5
+    epochs_between_evals: 5
+    # 默认参数1p 0.75 2p 1 4p 2 8p 4
+    lr: 0.75
+
+    # docker 镜像名称:版本号
+    docker_image: c73:b02
+
+    # 指定 device id, 数量需与 rank_size 相同
+    device_group_1p: 0
+    device_group_multi: 0,1,2,3,4,5,6,7
+