use OneFlow-Benchmark default optimizer

c5c01eec · ShawnXuan · 1b8c7c83 · c5c01eec
隐藏空白更改
内联并排

Showing with 0 addition and 1 deletion

OneFlow/scripts/rn50_train.sh OneFlow/scripts/rn50_train.sh +0 -1

未找到文件。
--- a/OneFlow/scripts/rn50_train.sh
+++ b/OneFlow/scripts/rn50_train.sh
@@ -24,7 +24,6 @@ python3 ./$BENCH_ROOT/of_cnn_train_val.py \
    --train_data_part_num=$DATA_PART_NUM \
    --num_nodes=$NUM_NODES \
    --gpu_num_per_node=$GPU_NUM_PER_NODE \
-    --model_update="momentum" \
    --learning_rate=0.001 \
    --loss_print_every_n_iter=20 \
    --batch_size_per_device=$BSZ_PER_DEVICE \