coco_hparams#

COCO (Common Objects in Context) dataset hyperparameters.

Hparams

These classes are used with yahp for YAML-based configuration.

COCODatasetHparams

Defines an instance of the COCO Dataset.

StreamingCOCOHparams

DatasetHparams for creating an instance of StreamingCOCO.
class composer.datasets.coco_hparams.COCODatasetHparams(drop_last=True, shuffle=True, is_train=True, datadir=None)[source]#

Bases: composer.datasets.dataset_hparams.DatasetHparams

Defines an instance of the COCO Dataset.

Parameters

  • datadir (str) โ€“ The path to the data directory.

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

class composer.datasets.coco_hparams.StreamingCOCOHparams(drop_last=True, shuffle=True, remote='s3://mosaicml-internal-dataset-coco/mds/1/', local='/tmp/mds-cache/mds-coco/', split='train')[source]#

Bases: composer.datasets.dataset_hparams.DatasetHparams

DatasetHparams for creating an instance of StreamingCOCO.

Parameters

  • remote (str) โ€“ Remote directory (S3 or local filesystem) where dataset is stored. Default: 's3://mosaicml-internal-dataset-coco/mds/1/`

  • local (str) โ€“ Local filesystem directory where dataset is cached during operation. Default: '/tmp/mds-cache/mds-coco/`

  • split (str) โ€“ The dataset split to use, either โ€˜trainโ€™ or โ€˜valโ€™. Default: 'train`.