| from src.data import make_loaders | |
| def main(): | |
| train_loader, val_loader, stoi = make_loaders(batch_size=4, img_height=64) | |
| batch = next(iter(train_loader)) | |
| print("Vocab size:", len(stoi)) | |
| print("Images:", batch.images.shape) # [B, 1, H, Wmax] | |
| print("Image widths:", batch.image_widths) # [B] | |
| print("Targets shape:", batch.targets.shape) # [sum(T)] | |
| print("Target lengths:", batch.target_lengths) | |
| print("Sample texts:", batch.texts[:2]) | |
| if __name__ == "__main__": | |
| main() | |