Change modality one-hot encoding to a learnt encoding

[jacobbieker]

Detailed Description

AS @JackKelly pointed out, the PerceiverIO paper uses a learnt embedding for the modality, while this uses a one-hot vector instead. Switching to a learnt embedding might be better?

Context

Possible Implementation

A torch.nn.Embedding with a vocab size set to the number of data sources