setting backend to torch (WIP)

brainglobe · Feb 9, 2024 · 5310e94 · 5310e94
1 parent 89339fa
commit 5310e94
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/cellfinder/__init__.py b/cellfinder/__init__.py
@@ -27,9 +27,9 @@
 # If no backend is configured and installed for Keras, tools cannot be used
 # Check backend is configured
 if not os.getenv("KERAS_BACKEND"):
-    os.environ["KERAS_BACKEND"] = "tensorflow"
+    os.environ["KERAS_BACKEND"] = "torch"  # "tensorflow"
     warnings.warn(
-        "Keras backend not configured, automatically set to Tensorflow"
+        "Keras backend not configured, automatically set to torch" 
     )
 
 # Check backend is installed

diff --git a/cellfinder/core/train/train_yml.py b/cellfinder/core/train/train_yml.py
@@ -451,6 +451,9 @@ def run(
         epochs=epochs,
         callbacks=callbacks,
     )
+    # model.built = True,
+    # model.compiled=True, and
+    # compile_metrics_unbuilt ---> I get error
 
     if save_weights:
         logger.info("Saving model weights")