diff --git a/train.py b/train.py
index 661dcf5..b70b2f1 100644
--- a/train.py
+++ b/train.py
@@ -24,7 +24,7 @@ from mindspore.context import ParallelMode
 from mindspore.communication.management import init, get_rank
 import time
 from OpenI import OpenIMultiDatasetToEnv as DatasetToEnv
-from OpenI import EnvOutputToOpenI,UploadOutputEpochEnd as EnvOutputToOpenI,UploadOutputEpochEnd
+from OpenI import UploadOutputEpochEnd as UploadOutputEpochEnd
 
 parser = argparse.ArgumentParser(description='MindSpore Lenet Example')