cerebras.modelzoo.data.vision.utils#
Functions
Checks for the dir in the worker_cache (SSD) on the worker node corresponding to the src_dir. |
|
Returns True if DDP is enabled |
|
Classes
Modified from: https://pytorch.org/docs/stable/_modules/torch/utils/data/distributed.html#DistributedSampler Sampler that restricts data loading to a subset of the dataset. |