updated or_imagenet dataset loader