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__": ...@@ -162,24 +162,24 @@ if __name__ == "__main__":
"--patch_size", "--patch_size",
type=int, type=int,
default=32, default=32,
help="the size of patches extracted for each reconstruction" help="the size of patches extracted for each reconstruction",
) )
parser.add_argument( parser.add_argument(
"--patch_offset", "--patch_offset",
type=int, type=int,
default=20, default=20,
help="offset to ignore the border of the image" help="offset to ignore the border of the image",
) )
parser.add_argument( parser.add_argument(
"--number_of_patches", "--number_of_patches",
type=int, type=int,
default=100, default=100,
help="number of patches extracted for each image" help="number of patches extracted for each image",
) )
parser.add_argument( parser.add_argument(
"--no_shuffle", "--no_shuffle",
action="store_true" action="store_true",
help="do not shuffle patches in the dataset" help="do not shuffle patches in the dataset",
) )
# Training Args # 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