Yichen Chi

bsz = hidden_states.shape[0] encoder_outputs = [] for splitted_hidden_states in hidden_states.split(bsz, dim=0): encoder_outputs_last = self.encoder(inputs_embeds=splitted_hidden_states) encoder_outputs.append(encoder_outputs_last) encoder_outputs = torch.cat(encoder_outputs) return encoder_outputs 我改成了这样，但还是会爆显存。。似乎并没有多少改善