From 4055fbdcc531ad59532a72c16a1b48a265b09174 Mon Sep 17 00:00:00 2001 From: zhang_xu_hao1230 Date: Mon, 9 Dec 2024 20:03:30 +0800 Subject: [PATCH] =?UTF-8?q?dp=E5=9F=9F=E5=90=88=E5=B9=B6=E4=BC=98=E5=8C=96?= =?UTF-8?q?=E5=99=A8=E6=B6=88=E9=99=A4bugfix?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- mindspeed_ms/tools/converter/combine_ckpt_dp_zero.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspeed_ms/tools/converter/combine_ckpt_dp_zero.py b/mindspeed_ms/tools/converter/combine_ckpt_dp_zero.py index 49122399..bcd6fd5d 100644 --- a/mindspeed_ms/tools/converter/combine_ckpt_dp_zero.py +++ b/mindspeed_ms/tools/converter/combine_ckpt_dp_zero.py @@ -77,7 +77,7 @@ def combine_zero3_data(param_total_dict, param_name, no_save_optim): return param_data def check_key(key): - key_list = ['learning_rate', 'weight_decay', 'epoch', 'state', 'default_generator'] + key_list = ['learning_rate', 'weight_decay', 'epoch', 'step', 'default_generator'] if any(x in key for x in key_list): return True return False -- Gitee