OpenNMT-tf icon indicating copy to clipboard operation
OpenNMT-tf copied to clipboard

Multi-decoder setup for multi-task learning

Open guillaumekln opened this issue 6 years ago • 5 comments

Similarly to the ParallelEncoder, a ParallelDecoder setup could allow multi-task learning. This should not be too hard to implement but we need to take care of some details:

  • support separate values for the decoding parameters (beam_width, length_penalty, etc.),
  • parts of SequenceToSequence assume a single output head (e.g. loss computation, reverse vocabulary lookup, exported outputs for model serving, etc. which should be moved in the decoder itself)

guillaumekln avatar May 22 '18 13:05 guillaumekln

@guillaumekln Do you have a plan to implement ParallelDecoder?

BigFishMaster avatar Jun 07 '19 06:06 BigFishMaster

I don't have plan to work on this at the moment.

guillaumekln avatar Jun 07 '19 07:06 guillaumekln

@guillaumekln is there any possibility on OpenNMT-py to integrate multi encoder decoders setup for multi-task learning? If so I wish to contribute for it.

sarubi avatar Aug 07 '20 06:08 sarubi

This issue is for OpenNMT-tf. If you wish to contribute the feature for OpenNMT-py, can you update the issue you opened there?

guillaumekln avatar Aug 07 '20 07:08 guillaumekln

@guillaumekln updated on OpenNMT-py.

sarubi avatar Aug 08 '20 08:08 sarubi