change lstm import pacakge location

model_zoo/lstm/eval.py

@@ -23,8 +23,8 @@ import numpy as np
 
 from src.config import lstm_cfg as cfg
 from src.dataset import lstm_create_dataset, convert_to_mindrecord
+from src.lstm import SentimentNet
 from mindspore import Tensor, nn, Model, context
-from mindspore.model_zoo.lstm import SentimentNet
 from mindspore.nn import Accuracy
 from mindspore.train.callback import LossMonitor
 from mindspore.train.serialization import load_checkpoint, load_param_into_net

model_zoo/lstm/train.py

@@ -24,8 +24,8 @@ import numpy as np
 from src.config import lstm_cfg as cfg
 from src.dataset import convert_to_mindrecord
 from src.dataset import lstm_create_dataset
+from src.lstm import SentimentNet
 from mindspore import Tensor, nn, Model, context
-from mindspore.model_zoo.lstm import SentimentNet
 from mindspore.nn import Accuracy
 from mindspore.train.callback import LossMonitor, CheckpointConfig, ModelCheckpoint, TimeMonitor
 from mindspore.train.serialization import load_param_into_net, load_checkpoint