保錯信息：RuntimeError: running_mean should contain 3 elements not 1
解決：因為模型的輸入是灰度圖像輸入，也就是輸入size為（batch_size,1,H,W）,可是我實際輸入模型的圖片是RGB三通道圖像，所以報錯，將RGB轉成灰度圖就好啦。
轉化灰度圖代碼
from torchvision import transforms transform = transforms.Compose([ transforms.Grayscale(num_output_channels=1), # 彩色圖像轉灰度圖像num_output_channels默認1 transforms.ToTensor() ])