请问如何在保存模型时,加入步数名的前缀
if step % 2 == 0:
ms.save_checkpoint(net,ckpt_file_name='D:\mindspore\models\\net.ckpt')
我尝试着
if step % 2 == 0:
ms.save_checkpoint(net,ckpt_file_name=step+'D:\mindspore\models\\net.ckpt')
但是会报错,请问如何解决?
您好,ckpt_file_name是一个路径名,猜想您想要的效果是存放许多个net_{step}.ckpt,建议将代码改成:
if step % 2 == 0:
ms.save_checkpoint(net,ckpt_file_name=f'D:\mindspore\models\net_{step}.ckpt')
试试看。