Files
ascend-tools/train/yaml/EfficientNet.yaml
T
2020-10-19 20:22:23 +08:00

21 lines
520 B
YAML
Raw Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
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可选 12或23 两两相连 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