diff --git a/src/utils_data.py b/src/utils_data.py index eea61c59be29003b17ae65f354064dd394628076..afa820cace1fa73ae6ee05e4c133494e191a108f 100644 --- a/src/utils_data.py +++ b/src/utils_data.py @@ -46,7 +46,6 @@ def create_label_dict(dataset : str, nn_model : str) -> dict: import sys import numpy as np import torchvision - from extra_keras_datasets import kmnist import torchvision.transforms as transforms transform = transforms.Compose(