【已解決】RuntimeError: stack expects each tensor to be equal size, but got [3, 256, 341] at entry 0

【報錯】如下：

RuntimeError: stack expects each tensor to be equal size, but got [3,256, 341] at entry 0 and [3, 256, 386] at entry 1

【分析】

train_dataset = datasets.ImageFolder(
    traindir,
    transforms.Compose([
        transforms.Resize((224)) ###
           

【原因】

原因是transforms.Resize() 的參數設定問題，改為如下設定就可以了

【解決】

train_dataset = datasets.ImageFolder(
    traindir,
    transforms.Compose([
        transforms.Resize((224,224)),
           

搞定！撒花…