Skip to content

GitLab

Commit 68e0eb47 authored by muyangli's avatar muyangli
Browse files

update

parent 57bac67d
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/utils.py
View file @ 68e0eb47
...@@ -59,7 +59,7 @@ class MultiImageDataset(data.Dataset): ...@@ -59,7 +59,7 @@ class MultiImageDataset(data.Dataset):
def compute_lpips( def compute_lpips(
ref_dirpath: str, gen_dirpath: str, batch_size: int = 4, num_workers: int = 1, device: str | torch.device = "cuda" ref_dirpath: str, gen_dirpath: str, batch_size: int = 4, num_workers: int = 0, device: str | torch.device = "cuda"
) -> float: ) -> float:
os.environ["TOKENIZERS_PARALLELISM"] = "false" os.environ["TOKENIZERS_PARALLELISM"] = "false"
metric = LearnedPerceptualImagePatchSimilarity(normalize=True).to(device) metric = LearnedPerceptualImagePatchSimilarity(normalize=True).to(device)
......
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