Merge branch 'main' into qcnn.py

SaashaJoshi · Dec 20, 2024 · 4c0d544 · 4c0d544
2 parents 84c89ca + b3e7203
commit 4c0d544
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -80,7 +80,7 @@ from piqture.data_loader.mnist_data_loader import load_mnist_dataset
 from piqture.embeddings.image_embeddings.ineqr import INEQR
 
 # Load MNIST dataset
-train_dataset, test_dataset = load_mnist_dataset()
+train_dataset, test_dataset = load_mnist_dataset(img_size=(8, 8))
 
 # Retrieve a single image from the dataset
 image, label = train_dataset[0]