From 3ddb004c7e79b9cdd56728fb72dff5f1d89694d6 Mon Sep 17 00:00:00 2001 From: kshitijrajsharma Date: Wed, 8 May 2024 17:45:47 +0545 Subject: [PATCH] Fix multimasks error in feedback --- hot_fair_utilities/training/train.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/hot_fair_utilities/training/train.py b/hot_fair_utilities/training/train.py index 4fcdcf5..37ab1e3 100644 --- a/hot_fair_utilities/training/train.py +++ b/hot_fair_utilities/training/train.py @@ -78,14 +78,14 @@ def run_feedback( epoch_size: int, batch_size: int, freeze_layers: bool = True, - multi_masks: bool = False, + multimasks: bool = False, ): assert os.path.exists(input_path), "Input Feedback Path Doesn't Exist" assert os.path.exists(feedback_base_model), "Feedback base Model Doesn't Exist" os.environ.update(os.environ) os.environ["RAMP_HOME"] = model_home print("Starting to prepare data for training") - split_training_2_validation(input_path, output_path, split_training_2_validation) + split_training_2_validation(input_path, output_path, multimasks) print("Data is ready for training") apply_feedback( @@ -94,7 +94,7 @@ def run_feedback( epoch_size, batch_size, freeze_layers, - multi_masks, + multimasks, ) final_accuracy, final_model_path = extract_highest_accuracy_model(output_path) return (final_accuracy, final_model_path)