liguge/Deep-Residual-Shrinkage-Networks-for-intelligent-fault-diagnosis-DRSN-

请教代码

Opened this issue · 2 comments

T0ng0715 commented

请问为什么在训练网络时DataLoader中batchsize只能为1不然会报错，但设为1后bn层用不了了。

T0ng0715 commented

Expected more than 1 value per channel when training, got input size torch.Size([1, 64]) 超过1就会报这个错误，可以解答以下嘛

yeyulege commented

Expected more than 1 value per channel when training, got input size torch.Size([1, 64]) 超过1就会报这个错误，可以解答以下嘛

Hi! Maybe you could try to set "drop_last = True" in dataloader to fix this issue.