lukasruff / Deep-SVDD-PyTorch

A PyTorch implementation of the Deep SVDD anomaly detection method
MIT License
698 stars 197 forks source link

[bug fix] handle cifar10 #7

Open montanier opened 5 years ago

montanier commented 5 years ago

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