From 6a7b974346c63d5619900fc71aa9aebacf66ef22 Mon Sep 17 00:00:00 2001 From: caojian05 Date: Tue, 23 Jun 2020 11:28:27 +0800 Subject: [PATCH] change lstm import pacakge location --- model_zoo/lstm/eval.py | 2 +- model_zoo/lstm/train.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/model_zoo/lstm/eval.py b/model_zoo/lstm/eval.py index a9b81199c..6d731fbd0 100644 --- a/model_zoo/lstm/eval.py +++ b/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 diff --git a/model_zoo/lstm/train.py b/model_zoo/lstm/train.py index 732655f1d..51ae12c68 100644 --- a/model_zoo/lstm/train.py +++ b/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 -- GitLab