Freeze modules bug?
https://github.com/espnet/espnet/blob/25862dd93081746ac623720f31f4f0f323c9750e/espnet/tts/pytorch_backend/tts.py#L331-L334

Do you think to freeze the module of the model, the line should be change into:
 if any(mod.startswith(key) for key in args.freeze_mods): 
?


diff --git a/espnet/tts/pytorch_backend/tts.py b/espnet/tts/pytorch_backend/tts.py
index 9d4c870d7..01395199e 100644
--- a/espnet/tts/pytorch_backend/tts.py
+++ b/espnet/tts/pytorch_backend/tts.py
@@ -330,7 +330,8 @@ def train(args):
     # freeze modules, if specified
     if args.freeze_mods:
         for mod, param in model.state_dict().items():
-            if any(key.startswith(mod) for key in args.freeze_mods):
+            if any(mod.startswith(key) for key in args.freeze_mods):
+                logging.info(f"{mod} is frozen not to be updated.")
                 param.requires_grad = False
 
     # Setup an optimizer
['espnet/tts/pytorch_backend/tts.py']