From 2c4dc8c66018074b4d9659b471fe829f13f07b2e Mon Sep 17 00:00:00 2001 From: "xiangpeng.wxp" Date: Wed, 30 Nov 2022 17:49:55 +0800 Subject: [PATCH] [to #42322933] nlp csanmt translation fix finetuning bug nlp csanmt translation fix finetuning bug Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10923166 * [to #42322933] nlp csanmt translation fix finetuning bug --- modelscope/models/nlp/csanmt/translation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modelscope/models/nlp/csanmt/translation.py b/modelscope/models/nlp/csanmt/translation.py index 4bac8e6d..657c26f4 100644 --- a/modelscope/models/nlp/csanmt/translation.py +++ b/modelscope/models/nlp/csanmt/translation.py @@ -391,7 +391,7 @@ class CsanmtForTranslation(Model): # Optimization trainable_vars_list = [ v for v in tf.compat.v1.trainable_variables() - if 'Shared_Semantic_Embedding' not in v.name + if 'Semantic_Embedding' not in v.name and 'mini_xlm_encoder' not in v.name ] grads_and_vars = opt.compute_gradients(