Skip to content

Commit

Permalink
Adding image size in data loader in README (#151)
Browse files Browse the repository at this point in the history 
adding image size in data loader
  • Loading branch information
@Shivansh20128
Shivansh20128 authored Dec 11, 2024
1 parent b656f79 commit b3e7203
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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]
Expand Down

0 comments on commit b3e7203

Please sign in to comment.