运行train_ae.py之后数据集改为CWRU，出现一个警告.请问怎么解决

[zhulewen1998]

UserWarning: Using a target size (torch.Size([4, 1, 1024])) that is different to the input size (torch.Size([4, 1024])). This will likely lead to incorrect results due to broadcasting. Please ensure they have the same size.