PyTorch implementation of the Perceiver model by Jaegle et al. from DeepMind (pdf
, abs
)
You can install pytorch-perceiver
via pip
:
pip install pytorch-perceiver
You
from pytorch_perceiver import Perceiver
net = Perceiver()
x = torch.rand(128, 128, 3)
y = net(x)
If you spot anything out of line, drop an Issue or raise a PR. All ideas welcome!