StreamingADE20K#

class streaming.vision.StreamingADE20K(*, remote=None, local=None, split=None, download_retry=2, download_timeout=60, validate_hash=None, keep_zip=False, epoch_size=None, predownload=None, partition_algo='orig', cache_limit=None, num_canonical_nodes=None, batch_size=None, shuffle=False, shuffle_algo='py1s', shuffle_seed=9176, shuffle_block_size=262144, joint_transform=None, transform=None, target_transform=None)[source]#

Implementation of the ADE20K dataset using StreamingDataset.

Parameters
  • remote (str, optional) – Remote path or directory to download the dataset from. If None, its data must exist locally. StreamingDataset uses either streams or remote/local. Defaults to None.

  • local (str, optional) – Local working directory to download shards to. This is where shards are cached while they are being used. Uses a temp directory if not set. StreamingDataset uses either streams or remote/local. Defaults to None.

  • split (str, optional) – Which dataset split to use, if any. If provided, we stream from/to the split subdirs of remote and local. Defaults to None.

  • download_retry (int) – Number of download re-attempts before giving up. Defaults to 2.

  • download_timeout (float) – Number of seconds to wait for a shard to download before raising an exception. Defaults to 60.

  • validate_hash (str, optional) – Optional hash or checksum algorithm to use to validate shards. Defaults to None.

  • keep_zip (bool) – Whether to keep or delete the compressed form when decompressing downloaded shards. If False, keep iff remote is local or no remote. Defaults to False.

  • epoch_size (int, optional) – Number of samples to draw per epoch balanced across all streams. If None, takes its value from the total number of underlying samples. Provide this field if you are weighting streams relatively to target a larger or smaller epoch size. Defaults to None.

  • predownload (int, optional) – Target number of samples to download per worker in advance of current sample. Workers will attempt to download ahead by this many samples during, but not before, training. Recommendation is to provide a value greater than per device batch size to ensure at-least per device batch size number of samples cached locally. If None, its value gets derived using per device batch size and number of canonical nodes max(batch_size, 256 * batch_size // num_canonical_nodes). Defaults to None.

  • cache_limit (int, optional) – Maximum size in bytes of this StreamingDataset’s shard cache. Before downloading a shard, the least recently used resident shard(s) may be evicted (deleted from the local cache) in order to stay under the limit. Set to None to disable shard eviction. Defaults to None.

  • partition_algo (str) – Which partitioning algorithm to use. Defaults to orig.

  • num_canonical_nodes (int, optional) –

    Canonical number of nodes for shuffling with resumption. The sample space is divided evenly according to the number of canonical nodes. The higher the value, the more independent non-overlapping paths the StreamingDataset replicas take through the shards per model replica (increasing data source diversity). Defaults to None, which is interpreted as 64 times the number of nodes of the initial run.

    Note

    For sequential sample ordering, set shuffle to False and num_canonical_nodes to the number of physical nodes of the initial run.

  • batch_size (int, optional) – Per-device batch size, the same as what is passed to the DataLoader. This affects how the dataset is partitioned over the workers and is necessary for deterministic resumption and optimal performance. Defaults to None.

  • shuffle (bool) – Whether to iterate over the samples in randomized order. Defaults to False.

  • shuffle_algo (str) – Which shuffling algorithm to use. Defaults to py1s.

  • shuffle_seed (int) – Seed for Deterministic data shuffling. Defaults to 9176.

  • shuffle_block_size (int) – Unit of shuffle. Defaults to 1 << 18.

  • joint_transform (callable, optional) – A function/transforms that takes in an image and a target and returns the transformed versions of both. Defaults to None.

  • transform (callable, optional) – A function/transform that takes in an image and returns a transformed version. Defaults to None.

  • target_transform (callable, optional) – A function/transform that takes in the target and transforms it. Defaults to None.

get_item(idx)[source]#

Get sample by global index, blocking to load its shard if missing.

Parameters

idx (int) – Sample index.

Returns

Tuple[Any, Any] – Sample data and label.