RuntimeError: CUDA out of memory. Tried to allocate 256.00 MiB (GPU 0; 4.00 GiB total capacity; 3.20 GiB already allocated; 0 bytes free; 3.23 GiB reserved in total by PyTorch) 2022-08-17 在 colab 上正常可以跑在程式, 搬到弱弱的…
[PyTorch] output with shape [1, 28, 28] doesn’t match the broadcast shape [3, 28, 28] 2021-06-04 會產生這一個錯誤,原因是圖片的格式造成,是彩色還…