Merge branch 'master' into dev

2023-05-03 13:15:00 +08:00 · 2023-05-03 13:15:00 +08:00 · 4e9de7df41
parent 8c4b2f9f6d 3f43199449
commit 4e9de7df41
1 changed files with 1 additions and 3 deletions
--- a/utils/init.py
+++ b/utils/init.py
@ -1,5 +1,4 @@
-import torch.cuda
-import torch.backends
+import torch

 def torch_gc(DEVICE):
    if torch.cuda.is_available():
@ -8,7 +7,6 @@ def torch_gc(DEVICE):
            torch.cuda.ipc_collect()
    elif torch.backends.mps.is_available():
        try:
-            import torch.mps
            torch.mps.empty_cache()
        except Exception as e:
            print(e)