提交 93d3d896 编写于 作者: S ShawnXuan

Merge branch 'of_cnn_e2e' of https://github.com/Oneflow-Inc/OneFlow-Benchmark into of_cnn_e2e

Conflicts:
	cnn_e2e/job_function_util.py
	cnn_e2e/resnet_model.py
rm -rf core.*
DATA_ROOT=/dataset/ImageNet/ofrecord
DATA_ROOT=/mnt/13_nfs/xuan/ImageNet/ofrecord
#DATA_ROOT=/dataset/ImageNet/ofrecord
#DATA_ROOT=/dataset/imagenet-mxnet
#python3 cnn_benchmark/of_cnn_train_val.py \
#gdb --args \
......@@ -9,13 +10,13 @@ DATA_ROOT=/dataset/ImageNet/ofrecord
--train_data_part_num=256 \
--val_data_dir=$DATA_ROOT/validation \
--val_data_part_num=256 \
--num_nodes=2 \
--num_nodes=1 \
--node_ips='11.11.1.13,11.11.1.14' \
--gpu_num_per_node=4 \
--optimizer="momentum-cosine-decay" \
--learning_rate=0.256 \
--loss_print_every_n_iter=20 \
--batch_size_per_device=32 \
--batch_size_per_device=64 \
--val_batch_size_per_device=125 \
--model="resnet50"
#--use_fp16 true \
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册