mirror of
https://gitee.com/mindspore/mindformers.git
synced 2025-12-06 19:42:57 +08:00
Compare commits
2 Commits
3988632900
...
r1.5.0-bet
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
8751a99019 | ||
|
|
686fd8058f |
@@ -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())
|
||||
|
||||
|
||||
Reference in New Issue
Block a user