折騰了一上午，看到別人寫的博客沒辦法解決我的問題。忽然想到靈感。
我要做的是，測試一下我自己寫的數據是否可以通過網絡輸出。但只有一個批量。
他會報如下錯誤
這里只需要把model狀態改成eval()模式就可以了，因為在train()狀態下是進行批量訓練的。
改之后的代碼為
免責聲明！
本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。
猜您在找
ValueError:Expected more than 1 value per channel when training, got input size torch.Size([1, 512, 1, 1])
pytorch報錯：ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1,512,1,1])
深度學習報錯：Expected more than 1 value per channel when training, got input size torch.Size([1, 256])
pytorch中使用多顯卡訓練以及訓練時報錯：expect more than 1 value per channel when training, got input size..
Pytorch錯誤Expected input batch_size (324) to match target batch_size (4) Log In
【pytorch報錯解決】expected input to have 3 channels, but got 1 channels instead
Library Publication 時遇到 "more than one library with package name" 錯誤的解決方法
FFMPEG more samples than frame size (avcodec_encode_audio2) 的解決方案
mysql運行報The server time zone value 'Öйú±ê׼ʱ¼ä' is unrecognized or represents more than one time zone的解決方法
[Vue warn]: Invalid prop: type check failed for prop "value". Expected String, Number, got Array 解決方法