liang_chaoming@huawei.com
21 lines
520 B
YAML
21 lines
520 B
YAML
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
|