mirror of https://github.com/coqui-ai/TTS.git
remove debug line
This commit is contained in:
parent
b76aaf8ad4
commit
8565c508e4
|
@ -193,7 +193,6 @@ def set_weight_decay(model, weight_decay, skip_list={"decoder.attention.v"}):
|
||||||
if not param.requires_grad:
|
if not param.requires_grad:
|
||||||
continue
|
continue
|
||||||
if len(param.shape) == 1 or name in skip_list:
|
if len(param.shape) == 1 or name in skip_list:
|
||||||
print(name)
|
|
||||||
no_decay.append(param)
|
no_decay.append(param)
|
||||||
else:
|
else:
|
||||||
decay.append(param)
|
decay.append(param)
|
||||||
|
|
Loading…
Reference in New Issue