aljaz/DALLE2-pytorch
1
1
Fork 0
mirror of https://github.com/lucidrains/DALLE2-pytorch.git synced 2025-12-20 02:04:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix readme dataloader example

Browse Source
This commit is contained in:
Phil Wang
2022-09-20 08:39:52 -07:00
committed by GitHub
parent 39f8b6cf16
commit b39653cf96
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@@ -1068,7 +1068,7 @@ dataloader = create_image_embedding_dataloader(
) )
for img, emb in dataloader: for img, emb in dataloader:
print(img.shape) # torch.Size([32, 3, 256, 256]) print(img.shape) # torch.Size([32, 3, 256, 256])
print(emb.shape) # torch.Size([32, 512]) print(emb["img"].shape) # torch.Size([32, 512])
# Train decoder only as shown above # Train decoder only as shown above
# Or create a dataset without a loader so you can configure it manually # Or create a dataset without a loader so you can configure it manually