在自建中文数据集上出现错误

[zi1zi]

感谢您出色的工作，我自己创建了一个中文数据集，运行时出现了以下错误
Traceback (most recent call last):
File "main.py", line 593, in
main()
File "main.py", line 581, in main
train(opts)
File "main.py", line 135, in train
attr_A_intensity = label_A * attr_A_data + (1 - label_A) * attr_A_embd
RuntimeError: The size of tensor a (17) must match the size of tensor b (37) at non-singleton dimension 1
出错的位置如图所示，图中两个print的输出分别为 torch.Size([64, 17])，torch.Size([64, 37])