remove extra linear layer when not needed (#485)
remove extra linear layer when not needed (#485)


facebookresearch/ParlAI,facebookresearch/ParlAI,facebookresearch/ParlAI,facebookresearch/ParlAI,facebookresearch/ParlAI