CoNT
CoNT copied to clipboard
[NeurIPS'22 Spotlight] Data and code for our paper CoNT: Contrastive Neural Text Generation
你好,对您的工作十分感兴趣,可以告知下关于运行配置和训练时长方面的问题吗
Eval时的报错
`File "xxx/Consistency_model/CoNT/model/model.py", line 192, in generate encoder_feature = self.affine_transformation(encoder_hidden_states, attention_mask) # batch x h File "/xxx/Consistency_model/CoNT/model/model.py", line 103, in affine_transformation trans_tmp = trans_tmp * padding_mask.unsqueeze(-1).float() RuntimeError: The size of tensor...
Hi, While playing around, I have found an error in the function `torch_bleu` which is used to rank batch-negatives and beam-positives. In `model.model.CoNTGenerator.torch_bleu` (line 47-70), there is an severe mistake...