pytorch save load

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/jacke121/article/details/84375495

import time

import torch.utils.data.distributed
from distributed.protocol import serialize, deserialize
import cv2
import torch
import torch.nn.functional as f
if __name__ == '__main__':
    for i in range(1000):
        x = torch.rand(100, 3, 352, 352)
        # torch.set_num_threads(3)
        start = time.time()

        torch.save(x, 'd:/img/'+str(i)+'.dat')
        aaa= torch.load('d:/img/'+str(i)+'.dat')
        print('time4', time.time() - start)  # 需要8ms左右。

100张图片，读写一共需要0.8s。