Upload folder using huggingface_hub

device.py

@@ -1,9 +1,7 @@
 import torch
 
-device: str
-
-if torch.cuda.is_available(): device = 'cuda'
-elif torch.mps.is_available(): device = 'mps'
-else: device = 'cpu'
+device = torch.accelerator.current_accelerator() or torch.device('cpu')
 
 print(f"Device set to {torch.Tensor().to(device).device}")
+
+__all__ = ['device']

test.py

@@ -5,8 +5,7 @@ import random
 def test(round):
     sumacc = 0
     lenacc = 0
-    pbar = trange(round)
-    for r in pbar:
+    for r in (pbar := trange(round)):
         x = random.randint(1, 100)
         y = random.randint(1, 100)
         pans = run([x, y])
@@ -14,7 +13,7 @@ def test(round):
         acc = 1 - abs(tans - pans) / tans
         sumacc += acc
         lenacc += 1
-        pbar.set_description_str(f"Round {r}, Average Accuracy = {sumacc / lenacc * 100}%")
+        pbar.set_description_str(f"Round {r}, Average Accuracy = {sumacc / lenacc * 100 : .6f}%")
     print(f"Final Accuracy: {sumacc / lenacc * 100}%")
 
 if __name__ == '__main__':