[add]上传训练benchmark by z00560161
This commit is contained in:
@@ -0,0 +1,20 @@
|
||||
pytorch_config:
|
||||
# 基本参数
|
||||
data_url: /home/imagenet/
|
||||
|
||||
epoches: 90
|
||||
# 1p 参数为512 2p 1024 4p 2048 8p为4096
|
||||
batch_size: 512
|
||||
|
||||
seed: 49
|
||||
|
||||
# 默认参数1p 0.4 2p 0.8 4p 1.6 8p 3.2
|
||||
lr: 0.4
|
||||
|
||||
#单P指定 卡 默认0卡
|
||||
device: 0
|
||||
|
||||
# 2p 4p 8p 修改 2p可选 1,2或2,3 两两相连 4p 可选'0,1,2,3'或'1,2,3,4'规则同2p 8p默认为'0,1,2,3,4,5,6,7'
|
||||
device_group: '0,1,2,3,4,5,6,7'
|
||||
# docker 镜像名称:版本号
|
||||
docker_image: c73:b021
|
||||
Reference in New Issue
Block a user