build_imagenet_dataloader

composer.datasets.build_imagenet_dataloader(datadir, global_batch_size, is_train=True, drop_last=True, shuffle=True, resize_size=- 1, crop_size=224, **dataloader_kwargs)

Builds an ImageNet dataloader.

Parameters

• datadir (str) – path to location of dataset.

• global_batch_size (int) – Global batch size.

• is_train (bool) – Whether to load the training data or validation data. Default: True.

• drop_last (bool) – whether to drop last samples. Default: True.

• shuffle (bool) – whether to shuffle the dataset. Default: True.

• resize_size (int, optional) – The resize size to use. Use -1 to not resize. Default: -1.

• size (crop) – The crop size to use. Default: 224.

• **dataloader_kwargs (Dict[str, Any]) – Additional settings for the dataloader (e.g. num_workers, etc.)