google / seq2seq

A general-purpose encoder-decoder framework for Tensorflow

Home Page:https://google.github.io/seq2seq/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Disable draw the attention plot

iiapache opened this issue · comments

commented

Instead of setting the dump_attention_no_plot parameter's value to "true",you should use "dump_plots: false" to disable the attention plot.
config example:
python -m bin.infer \
--tasks "
- class: DecodeText
- class: DumpAttention
params:
output_dir: $PRED_DIR/attention
dump_plots: false" \
--model_dir $MODEL_DIR \
--batch_size 1000 \
--input_pipeline "
class: ParallelTextInputPipeline
params:
source_files:
- $DEV_SOURCES" \
> ${PRED_DIR}/predictions_char_temp.txt