解决RuntimeError: running_mean should contain 36864 elements not 4096

NoSuchKey