环境: 鲲鹏920 + Ascend 910 + CentOS Linux release 7.9.2009 (AltArch) + python 3.7.5 + mindspore 2.1.0
在notebook中, 通过以下代码, 我已经成功的在训练中保存了模型
ckpt_config = CheckpointConfig(save_checkpoint_steps=1, keep_checkpoint_max=20)
ckpt_callback = ModelCheckpoint(prefix='resnet50', directory='./ckpts', config=ckpt_config)
model.train(epoch_size, dataset, callbacks=[LossMonitor(1), ckpt_callback], dataset_sink_mode=True)
在notebook中, 通过以下代码, 我已经成功的加载了前面保存的模型
model = resnet50()
param_dict = load_checkpoint('ckpts/best.ckpt')
load_param_into_net(model, param_dict)
相同的代码, 由 rank table 脚本方式运行时, 提示无法加载模型权值

请问什么原因如何解决
环境: 鲲鹏920 + Ascend 910 + CentOS Linux release 7.9.2009 (AltArch) + python 3.7.5 + mindspore 2.1.0
在notebook中, 通过以下代码, 我已经成功的在训练中保存了模型
ckpt_config = CheckpointConfig(save_checkpoint_steps=1, keep_checkpoint_max=20)
ckpt_callback = ModelCheckpoint(prefix='resnet50', directory='./ckpts', config=ckpt_config)
model.train(epoch_size, dataset, callbacks=[LossMonitor(1), ckpt_callback], dataset_sink_mode=True)
在notebook中, 通过以下代码, 我已经成功的加载了前面保存的模型
model = resnet50()
param_dict = load_checkpoint('ckpts/best.ckpt')
load_param_into_net(model, param_dict)
相同的代码, 由 rank table 脚本方式运行时, 提示无法加载模型权值
请问什么原因如何解决