mindspore的Print算子有两种模式,一种是直接打屏,还有一种是数据落盘,保存到二进制文件中
默认打印在屏幕上。也可以保存在文件中,通过 context 设置 print_file_path 参数。
一旦设置,输出将保存在指定文件中。通过函数 mindspore.parse_print() 可以重新加载数据。
而且print_file_path只在Ascend上设置生效
1. 保存print的数据到文件中
关键的代码为:context.set_context(print_file_path=print_file)
from mindspore import nn
from mindspore.ops import operations as ops
from minds