基於Pytorch實現Two stream early fusion的問題(vgg16)

小弟我想要實現two stream early fusion,我設計的網路架構如圖,但是跑執行訓練時出現了

RuntimeError: Given groups=1, weight of size [64, 3, 3, 3], expected input[32, 0, 224, 224] to have 3 channels, but got 0 channels instead

這樣子的錯誤,請教各位大神能否幫幫我以及給予這方面架構實現的相關建議

基於Pytorch實現Two stream early fusion的問題(vgg16)_第1张图片

基於Pytorch實現Two stream early fusion的問題(vgg16)_第2张图片 

 

你可能感兴趣的:(python,pytorch)