fix

Samsung · Jan 17, 2025 · d3b3c87 · d3b3c87
1 parent cae9cd5
commit d3b3c87
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/experiment/main.py b/experiment/main.py
@@ -306,11 +306,11 @@ def main(cred_data_location: str,
     subprocess.check_call(command, shell=True, cwd=pathlib.Path(__file__).parent)
 
     _model_file_name = main(cred_data_location=args.cred_data_location,
-                            jobs=args.jobs,
-                            epochs=args.epochs,
-                            batch_size=args.batch_size,
-                            patience=args.patience,
-                            doc_target=args.doc_target,
-                            use_tuner=args.use_tuner)
+                            jobs=int(args.jobs),
+                            epochs=int(args.epochs),
+                            batch_size=int(args.batch_size),
+                            patience=int(args.patience),
+                            doc_target=bool(args.doc_target),
+                            use_tuner=bool(args.use_tuner))
     # print in last line the name
     print(f"\nYou can find your model in:\n{_model_file_name}")