From b3577286a2e790d96115eb297fe37b496e4b950d Mon Sep 17 00:00:00 2001 From: root Date: Fri, 25 Dec 2020 17:03:28 +0800 Subject: [PATCH] lstm script fix --- model_zoo/official/nlp/lstm/eval.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model_zoo/official/nlp/lstm/eval.py b/model_zoo/official/nlp/lstm/eval.py index f7163fa0dee..4a9fa1d96f6 100644 --- a/model_zoo/official/nlp/lstm/eval.py +++ b/model_zoo/official/nlp/lstm/eval.py @@ -20,7 +20,7 @@ import os import numpy as np -from src.config import lstm_cfg as cfg, lstm_cfg_ascend +from src.config import lstm_cfg, lstm_cfg_ascend from src.dataset import lstm_create_dataset, convert_to_mindrecord from src.lr_schedule import get_lr from src.lstm import SentimentNet