tensor数据的保存--方便下次直接调用

最新推荐文章于 2024-08-25 13:36:07 发布

UndefindX-Z

最新推荐文章于 2024-08-25 13:36:07 发布

阅读量195

点赞数 1

文章标签：前端开发语言 tensor的保存

本文链接：https://blog.csdn.net/m0_75015083/article/details/137654692

版权

保存（假设count_0和count_1是两份tensor数据）：

import json

# 保存 count_0 和 count_1 到文件
with open('count_data.json', 'w') as f:
    json.dump({'count_0': count_0, 'count_1': count_1}, f)

后期使用：

import json

try:
    # 尝试从文件加载已保存的 count_0 和 count_1
    with open('count_data.json', 'r') as f:
        count_data = json.load(f)
        count_0 = count_data['count_0']
        count_1 = count_data['count_1']
        print("Loaded saved count data.")
except FileNotFoundError:
    # 如果文件不存在，执行原代码并保存结果
    for index, data in tqdm(enumerate(tbar)):
        # ...（此处省略原代码）

    # 保存 count_0 和 count_1 到文件（此处省略保存代码）

    print("Ran computation and saved count data.")