Force PYTORCH_MPS_FALLBACK to be set to 1 (#433)

brainglobe · Jun 3, 2024 · 093ba09 · 093ba09
1 parent 01f53e5
commit 093ba09
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/cellfinder/__init__.py b/cellfinder/__init__.py
@@ -26,6 +26,7 @@
 
 # Set the Keras backend to torch
 os.environ["KERAS_BACKEND"] = "torch"
+os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
 
 __license__ = "BSD-3-Clause"