Skip to content
Snippets Groups Projects
Commit 26bc3d50 authored by Tamino Huxohl's avatar Tamino Huxohl
Browse files

fix a bug in args definition for training

parent 6715925a
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment