PROSAGA码农传奇-pytorch-如何在Pytorch中保存和加载随机数生成器状态？

如何在Pytorch中保存和加载随机数生成器状态？

作者: 易米烊光
发布时间: 2024-12-11 12:12:10 (24天前)
转自：

<div class =“post-text”itemprop =“text”>
  
    您可以使用
    <a href="https://pytorch.org/docs/stable/torch.html#torch.get_rng_state" rel="nofollow noreferrer">
       <code>
 torch.get_rng_state
 </code>
    </A>
     和
    <a href="https://pytorch.org/docs/stable/torch.html#torch.set_rng_state" rel="nofollow noreferrer">
       <code>
 torch.set_rng_state
 </code>
    </A>
  
  
    打电话的时候
     <code>
 torch.get_rng_state
 </code>
     你会得到你的随机数发生器状态作为火炬.ByteTensor。
  
  
    然后，您可以将此张量保存在文件中的某个位置，以后可以加载和使用
     <code>
 torch.set_rng_state
 </code>
     设置随机数生成器状态。
  
  <HR />
  
    
      使用时
       <code>
 numpy
 </code>
       你当然可以在那里使用：
       
      <a href="https://docs.scipy.org/doc/numpy/reference/generated/numpy.random.get_state.html#numpy.random.get_state" rel="nofollow noreferrer">
         <code>
 numpy.random.get_state
 </code>
      </A>
       和
      <a href="https://docs.scipy.org/doc/numpy/reference/generated/numpy.random.set_state.html#numpy.random.set_state" rel="nofollow noreferrer">
         <code>
 numpy.random.set_state
 </code>
      </A>
    
  
</DIV>