torch序列化

同一张文件需要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(1000):
img = cv2.imread('