File size: 534 Bytes
2411029 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
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()
|