Example to compare datasets

[njan-creative]

Can you add an example to compare two datasets for a single model.

[MohammadJavadD]

I am using this code snippet but it seems something is wrong:

cka = CKA(model, model,
        model1_name="Model1 Data1", model2_name="Model1 Data2",
        device='cuda')

cka.compare(dataloader1, dataloader2)