From 6cbdbd8b32ebe7a77c2e46b1b38b7215f95d81c6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= Date: Wed, 31 Jan 2024 19:18:56 +0800 Subject: [PATCH] Update models.py --- models.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/models.py b/models.py index bd324b34b..ed3a2fee6 100644 --- 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)