Add multimasks in tasks

hotosm · May 1, 2024 · 01714e7 · 01714e7
1 parent 1b81dfd
commit 01714e7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/backend/core/tasks.py b/backend/core/tasks.py
@@ -231,6 +231,7 @@ def train_model(
                     model="ramp",
                     model_home=os.environ["RAMP_HOME"],
                     freeze_layers=freeze_layers,
+                    multimasks=multimasks
                 )
 
             # copy final model to output

diff --git a/backend/fAIr-utilities b/backend/fAIr-utilities