/perceiver

Primary LanguagePython

Implementation of the Perceiver model

This is an implementation "as i see it". Might not reflect the original implementation fully.

Usage

To run the a random tensor through the model.

python perceiver.py

To train on SST2

python train_sst2.py