Compare commits

...

2 Commits

Author SHA1 Message Date
i-robot
8751a99019 !5474 [r1.5.0-beta1]skip saving shared weight in last pipeline stage
Merge pull request !5474 from huangziling/r1.5.0-beta1
2025-02-24 07:45:08 +00:00
huangziling
686fd8058f skip saving shared weight in last pipeline stage 2025-02-22 09:44:59 +08:00

View File

@@ -929,6 +929,9 @@ class CheckpointMonitor(ModelCheckpoint):
param_dict[param.name] = param
param_list = []
for (key, value) in param_dict.items():
cur_param_info = value.param_info
if hasattr(cur_param_info, 'is_pipeline_shared_param') and cur_param_info.is_pipeline_shared_param:
continue
each_param = {"name": key}
param_data = Tensor(value.data.asnumpy())