0 votes
par (260 points) dans Séquence 14 : TP
Bonjour,

 En voulant exécuter le TP sur ma machine, j'ai le message d'erreur suivant:

dataset = load_dataset("/home/dcussol/Documents/fidle-tp/Diffusion_Model-TPs/fashion_mnist/")  # Use this one if you're not on Jean Zay
print ('Data Set infos...')
dataset

train_dataset = dataset['test']
print ('Train data set infos...')
train_dataset[0]

image = train_dataset[0]['image']
print ('Images infos...')
image
 

===> Message :

Traceback (most recent call last):
  File "/home/dcussol/Documents/fidle-tp/Diffusion_Model-TPs/TP_diff_model_dataset.py", line 43, in <module>
    image = train_dataset[0]['image']
KeyError: 'image'

Ou est mon erreur ?

Merci de votre aide...

Daniel Cussol

1 Réponse

0 votes
par (3.2k points)
sélectionné par
 
Meilleure réponse
Bonjour,

Vous pouvez essayer de charger le dataset avec la commande :
dataset = load_dataset('fashion_mnist')

L'architecture du dataset récupéré devrait alors être correcte.
...