Performs all_gather operation on the provided tensors. *** Warning ***: torch.distributed.all_gather has no gradient.

With 50% prob, negate the value

1x1 convolution

3x3 convolution with padding

Computes the accuracy over the k top predictions for the specified values of k

Returns list of utf-8 byte and a corresponding list of unicode strings. The reversible bpe codes work on unicode strings. This means you

3x3 convolution with padding

Return set of symbol pairs in a word. Word is represented as tuple of symbols (symbols being variable-length strings).

Checks if a file is an allowed extension. Args: filename (string): path to a file extensions (iterable of strings): extensions to

Create a RandAugment transform :param config_str: String defining configuration of random augmentation. Consists of multiple sections separat

dim: feature dimension (default: 128) K: queue size; number of negative keys (default: 65536) m: moco momentum of updating ke

Batch shuffle, for making use of BatchNorm. *** Only support DistributedDataParallel (DDP) model. ***

Undo batch shuffle. *** Only support DistributedDataParallel (DDP) model. ***

Finds the class folders in a dataset. Args: dir (string): Root directory path. Returns: tuple: (cla

Initialize slurm distributed training environment. If argument ``port`` is not specified, then the master port will be system environment vari

Momentum update of the key encoder

Input: im_q: a batch of query images im_k: a batch of key images Output: logits, targets

Decay the learning rate based on schedule

checkpoint forward

rand augment ops for RGB images

rand augment ops for CMC images (removing color ops)