diff --git a/shimmer/modules/gw_module.py b/shimmer/modules/gw_module.py index ea0d15c3..9c1922eb 100644 --- a/shimmer/modules/gw_module.py +++ b/shimmer/modules/gw_module.py @@ -419,7 +419,7 @@ def fuse( $$\\lambda_k = \\frac{M_3}{\\frac{c'_1}{a_k} + \\frac{c'_2}{b_k}}$$ where $$M_3 = \\frac{1}{\\sum_{i=1}^D - \\frac{1}{\\frac{c'_1}{a_i} + \\frac{c'_2}{b_i}}$$ + \\frac{1}{\\frac{c'_1}{a_i} + \\frac{c'_2}{b_i}}}$$ Args: x (`LatentsDomainGroupT`): the group of latent representation.