ValueError: Expected more than 1 value per channel when training, got input size [1, 16, 1, 1](解决方案)

NoSuchKey