From 156239eb295d79d6bdfefe816ab98fff825a5479 Mon Sep 17 00:00:00 2001
From: Flowingsun007 <Flowingsun007@163.com>
Date: Mon, 7 Sep 2020 08:56:36 +0800
Subject: [PATCH] refine

---
 Classification/cnns/of_cnn_train_val.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Classification/cnns/of_cnn_train_val.py b/Classification/cnns/of_cnn_train_val.py
index 6f7d3d5..0eff59d 100755
--- a/Classification/cnns/of_cnn_train_val.py
+++ b/Classification/cnns/of_cnn_train_val.py
@@ -17,7 +17,7 @@
 import math
 import oneflow as flow
 import ofrecord_util
-import optimizer_util_new
+import optimizer_util
 import config as configs
 from util import Snapshot, Summary, InitNodes, Metric
 from job_function_util import get_train_config, get_val_config
@@ -86,7 +86,7 @@ def TrainNet():
     outputs = {"loss": loss, "predictions": predictions, "labels": labels}
 
     # set up warmup,learning rate and optimizer
-    optimizer_util_new.set_up_optimizer(loss, args)
+    optimizer_util.set_up_optimizer(loss, args)
     return outputs