From 4564d5fcc206f26e700a2d5c8ff7f15b96ac8c01 Mon Sep 17 00:00:00 2001
From: oscarchaufour <101994223+oscarchaufour@users.noreply.github.com>
Date: Fri, 10 Nov 2023 11:04:37 +0100
Subject: [PATCH] Update read_cifar.py
---
read_cifar.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/read_cifar.py b/read_cifar.py
index b856ea4..12cd6b5 100644
--- a/read_cifar.py
+++ b/read_cifar.py
@@ -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)
--
GitLab