dot log fix for shared
@@ -315,6 +315,7 @@ class Seq2seqAgent(TorchAgent):
'optimizer_type': self.opt['optimizer'],
}
shared['metrics'] = self.metrics # do after numthreads check
+ if self.beam_dot_log is True:
shared['beam_dot_dir'] = self.beam_dot_dir
return shared
