From ccfe9bfa41f876c45429c4ce966aceb88677da79 Mon Sep 17 00:00:00 2001 From: blepping Date: Fri, 2 Feb 2024 00:36:47 -0700 Subject: [PATCH] Remove linear option from Deep Shrink as it does not actually work --- py/deepshrink.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/py/deepshrink.py b/py/deepshrink.py index f15f5ad..17f2497 100644 --- a/py/deepshrink.py +++ b/py/deepshrink.py @@ -13,7 +13,6 @@ class DeepShrinkBleh: "bilinear", "area", "bislerp", - "linear", ) @classmethod @@ -94,7 +93,7 @@ class DeepShrinkBleh: sigma_start = model.model.model_sampling.percent_to_sigma(start_percent) sigma_end = model.model.model_sampling.percent_to_sigma(end_percent) # Arbitrary number that should have good enough precision - pct_steps = 200 + pct_steps = 400 pct_incr = 1.0 / pct_steps sig2pct = tuple( model.model.model_sampling.percent_to_sigma(x / pct_steps)