search

photosynthesis-team / piq

Measures and metrics for image2image tasks. PyTorch.
Apache License 2.0
1.32k stars 114 forks source link

Incorrect return type on `piq.ssim` #378

Open adamtheturtle opened 5 months ago

adamtheturtle commented 5 months ago

Describe the bug

piq.ssim's return type is List[Tensor] but in the default case it returns Tensor.