NPU examples

@@ -114,6 +114,15 @@ def main():
         train_kwargs.update(cuda_kwargs)
         test_kwargs.update(cuda_kwargs)
 
+    datasets.MNIST.resources = [
+        (
+        'https://ossci-datasets.s3.amazonaws.com/mnist/train-images-idx3-ubyte.gz', 'f68b3c2dcbeaaa9fbdd348bbdeb94873'),
+        (
+        'https://ossci-datasets.s3.amazonaws.com/mnist/train-labels-idx1-ubyte.gz', 'd53e105ee54ea40749a09fcbcd1e9432'),
+        ('https://ossci-datasets.s3.amazonaws.com/mnist/t10k-images-idx3-ubyte.gz', '9fb629c4189551a2d022fa330f9573f3'),
+        ('https://ossci-datasets.s3.amazonaws.com/mnist/t10k-labels-idx1-ubyte.gz', 'ec29112dd5afa0611ce80d1b7f02629c')
+    ]
+
     transform=transforms.Compose([
         transforms.ToTensor(),
         transforms.Normalize((0.1307,), (0.3081,)),

@@ -2,7 +2,7 @@ import torch
 import torch.npu
 
 print("NPU available?", torch.npu.is_available())
-print("Current device", torch.npu.current_device())
+print("Current device:", torch.npu.current_device())
 device = torch.device("npu:0")
 # x = torch.randn(2, 2).npu()
 # y = torch.randn(2, 2).npu()