被气笑了
【cpu上报错】
return torch.embedding(weight, input, padding_idx, scale_grad_by_freq, sparse)
IndexError: index out of range in self
【gpu上报错】
RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling `cublasCreate(handle)`
还有一堆的block
【解决方案】
是因为我added_token.json增加了vocab,所以在config.json上需要修改一下vocab的size,由原本的21128增加13个