Apply Gaussian Blur to the PIL image.

Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks).

Convert a ``PIL Image`` or ``numpy.ndarray`` to tensor, or keep as is if already a tensor.

Computes pixAcc and mIoU metric scores

Track a series of values and provide access to smoothed values over a window or the global series average.

Kozachenko-Leonenko entropic loss regularizer from Sablayrolles et al. - 2018 - Spreading vectors for similarity search

Linear layer to train on top of frozen features

Gets knn of test features from all processes on a chunk of the train features Each rank gets a chunk of the train features as well as a chun

2D image to patch embedding: (B,C,H,W) -> (B,N,D) Args: img_size: Image size. patch_size: Patch token size. in_chans