From c5c01eec9e1b1a7bea14d250844e04b8a0cd7b83 Mon Sep 17 00:00:00 2001
From: ShawnXuan <xiexuanx2@gmail.com>
Date: Fri, 11 Sep 2020 11:45:27 +0800
Subject: [PATCH] use OneFlow-Benchmark default optimizer

---
 OneFlow/scripts/rn50_train.sh | 1 -
 1 file changed, 1 deletion(-)

diff --git a/OneFlow/scripts/rn50_train.sh b/OneFlow/scripts/rn50_train.sh
index 199c613..8b664e3 100755
--- 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 \
-- 
GitLab