Switched to pkl mnist set.