diff --git a/example/named_entity_recognition/README.md b/example/named_entity_recognition/README.md index 2b28b3b5039b..c914a6985dfe 100644 --- a/example/named_entity_recognition/README.md +++ b/example/named_entity_recognition/README.md @@ -15,4 +15,12 @@ To reproduce the preprocessed training data: To train the model: -- `$ cd src && python ner.py` \ No newline at end of file +- `$ cd src && python ner.py` + +To run inference using trained model: + +1. Recreate the bucketing module using `sym_gen` defined in `ner.py` +2. Loading saved parameters using `module.set_params()` + +Refer to the `test` function in the [Bucketing Module example](https://github.com/apache/incubator-mxnet/blob/master/example/rnn/bucketing/cudnn_rnn_bucketing.py) +and this [issue](https://github.com/apache/incubator-mxnet/issues/5008) on Bucketing Module Prediction \ No newline at end of file