RuntimeError: Expected target size [6, 1, 224, 224], got [6, 3, 224, 224]

RuntimeError: Expected target size [6, 1, 224, 224], got [6, 3, 224, 224]