Feat: freezing decoder option

2024-02-23 20:57:54 +09:00
parent 20b3326b79
commit ffdbb8e0d1
9 changed files with 130 additions and 19 deletions
--- a/preprocess_all.py
+++ b/preprocess_all.py
@@ -52,6 +52,11 @@ if __name__ == "__main__":
        action="store_true",
        help="Freeze style vector",
    )
+    parser.add_argument(
+        "--freeze_decoder",
+        action="store_true",
+        help="Freeze decoder",
+    )
    parser.add_argument(
        "--use_jp_extra",
        action="store_true",
@@ -84,6 +89,7 @@ if __name__ == "__main__":
        freeze_JP_bert=args.freeze_JP_bert,
        freeze_ZH_bert=args.freeze_ZH_bert,
        freeze_style=args.freeze_style,
+        freeze_decoder=args.freeze_decoder,
        use_jp_extra=args.use_jp_extra,
        val_per_lang=args.val_per_lang,
        log_interval=args.log_interval,