Update models.py

fishaudio · Jan 31, 2024 · 6cbdbd8 · 6cbdbd8
1 parent 123b836
commit 6cbdbd8
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/models.py b/models.py
@@ -389,8 +389,6 @@ def __init__(
             stochastic_sample_codes=True,
             threshold_ema_dead_code=2,
             use_cosine_sim = True,
-            orthogonal_reg_weight = 10,                 # in paper, they recommended a value of 10
-            orthogonal_reg_active_codes_only = False,
         )
         self.out_feature_net = nn.Linear(512, hidden_channels)