地址:https://parl.ai/docs/tutorial_torch_generator_agent.htm
Two major models in ParlAI inherit from TorchGeneratorAgent
: seq2seq and transformer. You can try the transformer with the example below:
parlai train_model --model transformer/generator --task convai2 --model-file /tmp/testtransformer --beam-size 5 --batchsize 16
原文:https://www.cnblogs.com/yanghh/p/14784785.html