Skip to content
Snippets Groups Projects
Commit f6d852ec authored by nomindcarry's avatar nomindcarry
Browse files

fix lstm_crf not reset is_training

parent bfeb203f
No related branches found
No related tags found
No related merge requests found
......@@ -62,6 +62,9 @@ class EvalCabllBack(TimeMonitor):
self.callback = F1(len(self.tags_to_index_map))
self._best_val_F1 = 0
def epoch_begin(self, run_context):
self.network.is_training = True
def epoch_end(self, run_context):
"""save .ckpt files"""
self.network.is_training = False
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment