liang_chaoming@huawei.com
34 lines
1.1 KiB
YAML
34 lines
1.1 KiB
YAML
tensorflow_config:
|
|
# 基本参数
|
|
data_url: /home/imagenet_TF/
|
|
max_epoches: 100
|
|
epochs_between_evals: 1
|
|
batch_size: 32
|
|
log_dir: ckpt
|
|
max_train_steps: 1000
|
|
|
|
# 1p参数
|
|
mode_1p: train # train、evaluate、train_and_evaluate三种模式,训练(train) 或训练评测(train_and_evaluate)
|
|
iterations_per_loop_1p: 10
|
|
display_every: 10
|
|
log_name_1p: vgg16_1p.log
|
|
|
|
# 8p参数
|
|
mode_8p: train_and_evaluate # train、evaluate、train_and_evaluate三种模式,训练(train) 或训练评测(train_and_evaluate)
|
|
iterations_per_loop_8p: 5004
|
|
lr: 0.01
|
|
log_name_8p: vgg16_8p.log
|
|
|
|
# 仅多机执行需要配置: ip1:卡数量1,ip2:卡数量2
|
|
mpirun_ip: 90.90.176.154:8,90.90.176.54:8
|
|
|
|
# docker 镜像名称:版本号
|
|
docker_image: mpirun3:latest
|
|
|
|
# 1. 指定 device id, 多个 id 使用空格分隔, 数量需与 rank_size 相同
|
|
# 2. 仅在小于 8p 时生效
|
|
# 3. 若不使用该配置, 请使用在行首添加'#'注释的方法将其关闭
|
|
# device_group: 0 1 2 3
|
|
device_group_1p: 0
|
|
device_group_2p: 0 1
|
|
device_group_4p: 0 1 2 3 |