refine

Oneflow-Inc · Sep 7, 2020 · 156239e · 156239e
1 parent 638dc17
commit 156239e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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