This seems to give better quality in tome.

main
comfyanonymous 2 years ago
parent 18a6c1db33
commit 0d972b85e6

@ -109,7 +109,7 @@ def get_functions(x, ratio, original_shape):
w = original_w // downsample
h = original_h // downsample
r = int(x.shape[1] * ratio)
no_rand = True
no_rand = False
m, u = bipartite_soft_matching_random2d(x, w, h, stride_x, stride_y, r, no_rand)
return m, u

Loading…
Cancel
Save