Computes and stores the average and current value

pretrained MaskRCNN from torchvision

`MS Coco Detection <https://cocodataset.org/#detection-2016>`_ Dataset. It requires the `COCO API to be installed <https://github.com/pdollar/coc

a few conv layers to flatten features that come out of ResNet

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

Learned additive dataset id encoding implemented on top of nn.Embedding

Relative position in an instruction (a sentence) encoding with wavelets

object classifier module (a single FF layer)

Transformer-style positional encoding with wavelets

Transformer-style positional encoding with wavelets

Learned additive positional encoding implemented on top of nn.Embedding

pretrained FasterRCNN or MaskRCNN from torchvision

pretrained Resnet18 from torchvision

pretrained Resnet18 from torchvision

stream object that redirects writes to a logger instance.

Learned additive img/word/action token encoding implemented on top of nn.Embedding

Allow nn.DataParallel to call model's attributes.

dot-attention (or soft-attention)

Learned additive encoding implemented on top of nn.Embedding

self-attention with learnable parameters

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