search

ZhaoZhibin / UDTL

Source codes for the paper "Applications of Unsupervised Deep Transfer Learning to Intelligent Fault Diagnosis: A Survey and Comparative Study" published in TIM
MIT License
400 stars 103 forks source link

50 middle epoch #7

Open wb1997-cell opened 2 years ago

wb1997-cell commented 2 years ago

When I used the CWRU data set, using the adversarial method, THERE would be 50 epoch errors ValueError: Using a target size (torch.Size([128])) that is different to the input size (torch.Size([128, 1])) is deprecated. Please ensure they have the same size. What can I do to fix it that doesn't happen when I use SEU data sets

ZhaoZhibin commented 2 years ago

When I used the CWRU data set, using the adversarial method, THERE would be 50 epoch errors ValueError: Using a target size (torch.Size([128])) that is different to the input size (torch.Size([128, 1])) is deprecated. Please ensure they have the same size. What can I do to fix it that doesn't happen when I use SEU data sets

This is the problem with the version of Pytorch. I have fixed this issue. You can download the codes again.