I do not see any instructions on how to update code… can you be more specific?

  
      
      GitHub
  
  
    

pytorch/pytorch 5

Tensors and Dynamic neural networks in Python  with strong GPU acceleration - pytorch/pytorch