Browse Source

!2487 change lstm import pacakge location

Merge pull request !2487 from caojian05/mindspore_master_bugfix
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
8b7e8262a9
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      model_zoo/lstm/eval.py
  2. +1
    -1
      model_zoo/lstm/train.py

+ 1
- 1
model_zoo/lstm/eval.py View File

@@ -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


+ 1
- 1
model_zoo/lstm/train.py View File

@@ -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


Loading…
Cancel
Save