def add_to_argparse(parser):
BaseDataModule.add_to_argparse(parser)
parser.add_argument("--num_samples", type=int, default=_NUM_SAMPLES)
parser.add_argument("--num_classes", type=int, default=_NUM_CLASSES)
parser.add_argument("--image_height", type=int, default=_IMAGE_LEN)
parser.add_argument("--image_width", type=int, default=_IMAGE_LEN)
return parser
def add_to_argparse(parser):
BaseDataModule.add_to_argparse(parser)
parser.add_argument("--max_length", type=int, default=MAX_LENGTH, help="Max line length in characters.")
parser.add_argument(
"--min_overlap",
type=float,
default=MIN_OVERLAP,
help="Min overlap between characters in a line, between 0 and 1.",
)
parser.add_argument(
"--max_overlap",
type=float,
default=MAX_OVERLAP,
help="Max overlap between characters in a line, between 0 and 1.",
)
parser.add_argument("--with_start_end_tokens", action="store_true", default=False)
return parser
def add_to_argparse(parser):
BaseDataModule.add_to_argparse(parser)
parser.add_argument("--augment_data", type=str, default="true")
parser.add_argument("--max_length", type=int, default=MAX_LENGTH, help="Max line length in characters.")
parser.add_argument(
"--min_overlap",
type=float,
default=MIN_OVERLAP,
help="Min overlap between characters in a line, between 0 and 1.",
)
parser.add_argument(
"--max_overlap",
type=float,
default=MAX_OVERLAP,
help="Max overlap between characters in a line, between 0 and 1.",
)
return parser
def add_to_argparse(parser):
BaseDataModule.add_to_argparse(parser)
parser.add_argument("--augment_data", type=str, default="true")
return parser