sciai.utils

工具模块,包含常用Python工具、MindSpore工具和画图工具等。

sciai.utils.calc_ckpt_name

根据命名空间参数名拼接最优checkpoint的文件名。

sciai.utils.str2datatype

从浮点精度字符串到 MindSpore 数据类型的映射。

sciai.utils.amp2datatype

从自动混合精度字符串到 MindSpore 数据类型的映射。

sciai.utils.datatype2np

MindSpore 数据类型到 numpy 数据类型的映射。

sciai.utils.download_resource

为名为 model_name 的模型下载数据集与(或)checkpoints文件。

sciai.utils.flatten_add_dim

展平数据并在末尾增加额外一个维度。

sciai.utils.lazy_func

制造一个可以懒加载的函数,该函数可在之后直接被无参调用。

sciai.utils.lazy_property

惰性属性装饰器。

sciai.utils.log_config

日志配置。

sciai.utils.make_sciai_dirs

为 sciai 项目创建目录。

sciai.utils.newfig

绘制一个新的图片。

sciai.utils.parse_arg

根据终端/bash输入和config字典解析参数。

sciai.utils.print_args

将参数逐行打印与记录进日志。

sciai.utils.print_log

在标准输出流和日志文件中打印。

sciai.utils.print_time

打印函数端到端运行时间。

sciai.utils.savefig

pdfpng 格式保存图形。

sciai.utils.save_result_dir

将图片结果保存在指定目录中。

sciai.utils.set_seed

设置各种地方的随机种子。

sciai.utils.time_second

获取以毫秒为单位的时间,例如 1678243339.780746。

sciai.utils.time_str

获取时间的字符串表示,例如:"2000-12-31-23-59-59"。

sciai.utils.to_float

将若干 Cell 转换为指定的数据类型。

sciai.utils.to_tensor

将数组/张量转换为给定的MindSpore数据类型。

sciai.utils.to_tuple

如果不是元组,则将输入参数转换为元组,否则保持不变。