Browse Source

lstm script fix

tags/v1.1.0
root 5 years ago
parent
commit
b3577286a2
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      model_zoo/official/nlp/lstm/eval.py

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

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


Loading…
Cancel
Save