I’m not sure if I’m understanding this correctly, but if I change the datasets feature names to inputs (instead of embeddings) I’m still getting the same error.
class CLIPDataset(nn.utils.data.Dataset):
    def __init__(self, inputs, labels):
        self.inputs = inputs
        self.labels = labels

    def __getitem__(self, idx):
        item = {"inputs": nn.Tensor(self.embeddings[idx])}
        item['labels'] = nn.LongTensor([self.labels[idx]])
        return item