From 26bc3d50370d728743ccc2535768a164039ad937 Mon Sep 17 00:00:00 2001
From: Tamino Huxohl <thuxohl@techfak.uni-bielefeld.de>
Date: Wed, 5 Oct 2022 10:28:50 +0200
Subject: [PATCH] fix a bug in args definition for training
---
mu_map/training/default.py | 10 +++++-----
1 file changed, 5 insertions(+), 5 deletions(-)
diff --git a/mu_map/training/default.py b/mu_map/training/default.py
index fd554ff..8e09220 100644
--- a/mu_map/training/default.py
+++ b/mu_map/training/default.py
@@ -162,24 +162,24 @@ if __name__ == "__main__":
"--patch_size",
type=int,
default=32,
- help="the size of patches extracted for each reconstruction"
+ help="the size of patches extracted for each reconstruction",
)
parser.add_argument(
"--patch_offset",
type=int,
default=20,
- help="offset to ignore the border of the image"
+ help="offset to ignore the border of the image",
)
parser.add_argument(
"--number_of_patches",
type=int,
default=100,
- help="number of patches extracted for each image"
+ help="number of patches extracted for each image",
)
parser.add_argument(
"--no_shuffle",
- action="store_true"
- help="do not shuffle patches in the dataset"
+ action="store_true",
+ help="do not shuffle patches in the dataset",
)
# Training Args
--
GitLab