Skip to content

GitLab

Unverified Commit 945f3a8b authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

[FBcode->GH] change image type for LSUN tests (#3492)

parent 8a820172
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
test/test_datasets.py
View file @ 945f3a8b
...@@ -988,7 +988,7 @@ class LSUNTestCase(datasets_utils.ImageDatasetTestCase): ...@@ -988,7 +988,7 @@ class LSUNTestCase(datasets_utils.ImageDatasetTestCase):
folder = f"{cls}_lmdb" folder = f"{cls}_lmdb"
num_images = torch.randint(1, 4, size=()).item() num_images = torch.randint(1, 4, size=()).item()
format = "webp" format = "png"
files = datasets_utils.create_image_folder(root, folder, lambda idx: f"{idx}.{format}", num_images) files = datasets_utils.create_image_folder(root, folder, lambda idx: f"{idx}.{format}", num_images)
with lmdb.open(str(root / folder)) as env, env.begin(write=True) as txn: with lmdb.open(str(root / folder)) as env, env.begin(write=True) as txn:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment