Skip to content
Snippets Groups Projects
Commit 4564d5fc authored by oscarchaufour's avatar oscarchaufour
Browse files

Update read_cifar.py

parent 7dfd0b60
Branches main
No related tags found
No related merge requests found
......@@ -24,7 +24,7 @@ def read_cifar (batch_dir) :
data_batches = []
label_batches = []
for i in range(1,2) :
for i in range(1, 6) :
batch_filename = f'data_batch_{i}'
batch_path = os.path.join(batch_dir, batch_filename)
data, labels = read_cifar_batch(batch_path)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment