Update cifar_util.py