[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/zhusiyi/dataset/peta/
+    
+    epoches: 150
+    # 1p 参数为256 2p 512 4p 1024  8p为2048
+    batch_size: 256
+    
+    seed: 49
+    
+    # 默认参数1p 0.01 2p 0.016 4p 0.016 8p 0.016
+    lr: 0.016
+   
+    #单P指定 卡 默认7卡
+    device: 7
+   
+    # docker 镜像名称:版本号
+    docker_image: c73:b021