Update train.py

janelia-cellmap · Feb 7, 2024 · 8af4f90 · 8af4f90
1 parent 6b4e17a
commit 8af4f90
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dacapo/train.py b/dacapo/train.py
@@ -12,7 +12,7 @@
 logger = logging.getLogger(__name__)
 
 
-def train(run_name: str, compute_context: ComputeContext = LocalTorch(), force_cuda = False):
+def train(run_name: str, compute_context: ComputeContext = LocalTorch()):
     """Train a run"""
 
     if compute_context.train(run_name):