I’m trying DGL implementation of PinSAGE. I see that examples included in repo use text embeddings that are created with torchtext. I prefer to use image embeddings of items instead - is that possible?
I tried to do this in the processing part with that:
g.nodes['item'].data['image_embedding'] = torch.FloatTensor(list(items['image_embedding'].values))
I can see that model creates LinearProjector layer for these embeddings, but I’m not sure if this is a correct way to handle this. Do you have any other proposition for that?