down into cpp

pytorch · Apr 24, 2024 · 1b81829 · 1b81829
1 parent 67b5840
commit 1b81829
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/test_test_mnist.py b/test/test_test_mnist.py
@@ -13,6 +13,7 @@
 torch.manual_seed(random_seed)
 
 ### load data
+import torch_xla.utils.utils as xu
 test_loader = xu.SampleGenerator(
     data=(torch.zeros(flags.batch_size, 1, 28,
                       28), torch.zeros(flags.batch_size,