s424714/ium_424714
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix: jenkinsfile

Browse Source
This commit is contained in:
Filip Patyk 2023-05-10 17:07:51 +02:00
parent c9adc79c48
commit 26c57cf6b1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/main.py
View File

@ -21,7 +21,7 @@ parser.add_argument("--test", action="store_true", default=False)
parser.add_argument("--model_path", type=str, default="results/model.pt")
parser.add_argument("--results_path", type=str, default="results/results.csv")
parser.add_argument("--data_acc_path", type=str, default="./results/acc.csv")
parser.add_argument("--build_id", type=int, default=0)
parser.add_argument("--build_id", type=str, default="0")
# HYPER PARAMETERS
parser.add_argument("--batch", "-b", type=int, default=2)
@ -86,6 +86,6 @@ if __name__ == "__main__":
utils.save_results(labels=test_data["label"], results=results, file_path=args.results_path)
utils.save_data_to_csv(
file_path=args.data_acc_path,
build_id=args.build_id,
build_id=int(args.build_id),
data=accuracy,
)