We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 99178ef commit 570d557Copy full SHA for 570d557
1 file changed
main_training_mamba.py
@@ -113,7 +113,6 @@ def main(**kwargs):
113
params_with_decay = []
114
params_without_decay = []
115
for name, param in model.named_parameters():
116
- print(f'{name=}')
117
suff = name.split('.')[-1]
118
if 'A_log' in suff or 'D' in suff or 'dt_bias' in suff:
119
params_without_decay.append(param)
0 commit comments