pytorch save load

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。

读取需要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())
     
    
  
 

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/84375495