Browse Source

[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
master^2
xiangpeng.wxp mulin.lyh 3 years ago
parent
commit
2c4dc8c660
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      modelscope/models/nlp/csanmt/translation.py

+ 1
- 1
modelscope/models/nlp/csanmt/translation.py View File

@@ -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(


Loading…
Cancel
Save