train

KibromBerihu · Jun 24, 2022 · d8d8194 · d8d8194
1 parent f4eafad
commit d8d8194
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/LFBNet/utilities/train_valid_paths.py b/src/LFBNet/utilities/train_valid_paths.py
@@ -177,7 +177,7 @@ def get_output_or_create_folder_name(
 
     assert model in ['2D', "3D"]
 
-    # check if the directory to the proceseed data is given
+    # check if the directory to the processed data is given
     directory_exist(processed_data_directory)
     # if it exists
     data_dir = processed_data_directory

diff --git a/src/train.py b/src/train.py
@@ -128,7 +128,7 @@ def main():
 
         # train or test on the given input arguments
         trainer_params = dict(folder_preprocessed_train=dir_mip, folder_preprocessed_valid=dir_mip,
-                              ids_to_read_train=train_ids, ids_to_read_valid=valid_ids, training=task,
+                              ids_to_read_train=train_ids, ids_to_read_valid=valid_ids, task=task,
                               predicted_directory=output_dir)
 
         network_run = trainer.NetworkTrainer(**trainer_params)
@@ -149,7 +149,7 @@ def main():
             train_ids, valid_ids = train_valid_paths.get_train_valid_ids_from_folder(path_train_valid=path_train_valid)
 
         trainer_params = dict(folder_preprocessed_train=dir_mip, folder_preprocessed_valid=dir_mip,
-                              ids_to_read_train=train_ids, ids_to_read_valid=valid_ids, training=task,
+                              ids_to_read_train=train_ids, ids_to_read_valid=valid_ids, task=task,
                               predicted_directory=output_dir, save_predicted=True)
 
         network_run = trainer.NetworkTrainer(**trainer_params)