ocr / train_emnist.py
hanz245's picture
clean version without LFS
091afb2
Raw
History Blame
348 Bytes
import torchvision
import torchvision.transforms as transforms
print("Loading EMNIST dataset...")
train_data = torchvision.datasets.EMNIST(
root='datasets/emnist',
split='byclass',
train=True,
download=False,
transform=transforms.ToTensor()
)
print(f"Training samples: {len(train_data)}")
print("EMNIST loaded successfully!")