使用MGAN模型报错
cheemsbaby opened this issue · comments
File "D:\代码\文本\ABSA-PyTorch\models\mgan.py", line 56, in forward
alignment_mat[:, i, j] = feat.matmul(self.w_u.expand(batch_size, -1, -1)).squeeze(-1).squeeze(-1)
RuntimeError: Expected object of scalar type Double but got scalar type Float for argument #2 'mat2' in call to _th_bmm_out