Skip to content
This repository has been archived by the owner on Dec 29, 2022. It is now read-only.

Disable draw the attention plot #348

Open
iiapache opened this issue Jan 3, 2019 · 0 comments
Open

Disable draw the attention plot #348

iiapache opened this issue Jan 3, 2019 · 0 comments

Comments

@iiapache
Copy link

iiapache commented Jan 3, 2019

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

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant