CoNT
CoNT copied to clipboard
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 a (128) must match the size of tensor b (16) at non-singleton dimension 0
使用的模型为fnlp/cpt-large
,在运行训练时,如果在训练过程中需要eval,则会出现如上的错误,请问是什么问题?