ValueError：不推薦使用與輸入大小 (torch.Size([13456, 1])) 不同的目標大小 (torch.Size([16]))

Question

我的代碼： https://colab.research.google.com/drive/1qjfy2OsHYewhHDej-W83CMNercB7o7r8?usp=sharing

錯誤：ValueError：不推薦使用與輸入大小 (torch.Size([13456, 1])) 不同的目標大小 (torch.Size([16]))。 請確保它們的尺寸相同。

該數據集由 2 個文件夾組成：0 和 1，在這兩個文件夾中的每一個中，大約有 2500 張 512*512 圖像和每個圖像的 json 文件。

代碼來自 pytorch gan 教程，我只是更改了數據集。

我想知道 13456 從哪里來？

Answer 1

原始代碼適用於 64 x 64 圖像，而不是 512 x 512 圖像。 要解決此問題，您必須將圖像縮小到 64 x 64 或修改鑒別器和生成器。

Answer 2

label = torch.full((b_size,), real_label, dtype=torch.float, device=device)

這將創建一個大小為 b_size (16) 的數組，其中填充了 real_label (1)。

output = netD(real_cpu).view(-1)
output = output.unsqueeze(1)

output 的大小為 (x,1)，其中 x 是最后一個卷積網絡中的輸出數量，不等於批量大小。