RuntimeError: shape '[64, 1, 1, 1]' is invalid for input of size 6400

pytorch報錯RuntimeError: shape ‘[64, 1, 1, 1]’ is invalid for input of size 6400

源代碼

samples = samples.data.reshape(batch_size, 1, 1, 1)
    samples = samples.data.expand(batch_size, input_dim, 1, 1)
    samples = samples.cuda() if use_cuda else samples

原因:

由於修改了resize -> reshape,導致第一次循環對samples的改變維持到了第二輪循環

修改

把後面的samples -> samples1 即可

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章