RuntimeError: output with shape [1, 224, 224] doesn't match the broadcast shape [3, 224, 224]
ERROR : RuntimeError: output with shape [1, 224, 224] doesn't match the broadcast shape [3, 224, 224] 해결 : # Define transforms transform = transforms.Compose([ transforms.Resize((224, 224)), # Resize to the size the model expects transforms.ToTensor(), # ORG # transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]), # ORG to TUNE # TUNE transforms.Normalize((0.5), (0.5)) ]) 참..