update infer.
This commit is contained in:
parent
a0786c410b
commit
6a882aa35a
|
@ -12,7 +12,7 @@ import cged_config
|
||||||
from corpus_reader import CGEDReader
|
from corpus_reader import CGEDReader
|
||||||
from reader import EOS_ID
|
from reader import EOS_ID
|
||||||
from utils.text_utils import segment
|
from utils.text_utils import segment
|
||||||
from seq2seq.train import create_model
|
from train import create_model
|
||||||
|
|
||||||
|
|
||||||
def decode(sess, model, data_reader, data_to_decode,
|
def decode(sess, model, data_reader, data_to_decode,
|
||||||
|
|
Loading…
Reference in New Issue