torch序列化

最新推荐文章于 2024-08-27 23:43:52 发布

AI算法网奇

最新推荐文章于 2024-08-27 23:43:52 发布

阅读量2.6k

点赞数

CC 4.0 BY-SA版权

分类专栏： pytorch知识宝典

本文链接：https://blog.youkuaiyun.com/jacke121/article/details/84342788

pytorch知识宝典专栏收录该内容

512 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

本文介绍了如何使用PyTorch进行序列化操作，通过实例展示了Tensor的序列化与反序列化过程，确保数据在转换后的一致性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

同一张文件需要7ms左右：

import time
import torch
import torch.nn.functional as f
if __name__ == '__main__':
    for i in range(200):
        x = torch.rand(1, 3, 1280, 720)
       # torch.set_num_threads(3)

        torch.save(x, 'd:/lib/'+str(0)+'.dat')
        start = time.time()
        aaa= torch.load('d:/lib/'+str(0)+'.dat')
        print('time4', time.time() - start,aaa.size())

import time from distributed.protocol import serialize, deserialize import cv2 import torch import torch.nn.functional as f if __name__ == '__main__': obj={'mat':torch.randn(10, 10),'name': '10','test':{'entry':1}} torch.save(obj,'test.dat' ) for i in range(100