I’m not yet ran all of code.
I ran just ‘imshow’
def imshow(inp, img_num):
    inp = inp.numpy().transpose((1, 2, 0))
    mean = np.array([0.485, 0.456, 0.406])
    std = np.array([0.229, 0.224, 0.225])
    inp = std * inp + mean
    inp = np.clip(inp, 0, 1)
    print('label : {}'.format(train[img_num][1]))
    plt.imshow(inp)

imshow(train[40][0], 40)

But output shows mismatched label with pic