I compiled the codes by modify the calling, the codes uploaded to the git. If you are interest in, the follows is the Codes:

  
      
      GitHub
  
  
    

TonyzBi/libtorch-mnist 67

Contribute to TonyzBi/libtorch-mnist development by creating an account on GitHub.