With the provided coded, the example on cifar10 was failing. It seems that the cifar10 dataset does not have "train_labels" and "train/test_data" but respectively "targets" and "data".
This fix proposes a name change at key places to be able to run the example
With the provided coded, the example on cifar10 was failing. It seems that the cifar10 dataset does not have "train_labels" and "train/test_data" but respectively "targets" and "data".
This fix proposes a name change at key places to be able to run the example