diff --git a/backend/core/tasks.py b/backend/core/tasks.py index f8726a4e..7dcd32d5 100644 --- 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 new file mode 160000 index 00000000..93debb40 --- /dev/null +++ b/backend/fAIr-utilities @@ -0,0 +1 @@ +Subproject commit 93debb40704c94fb4f89ca93f660e8b5c0fdb747