class ConfigPostProcessFlags:
jq_transform: List[str] = cmd_line_flags.flag(
"Allows to change the configuration from the command line using jq "
"transformations. Applied before string template substitution.",
short_name="jq")
string_def: List[str] = cmd_line_flags.flag(
"Allows to replace the string template in the configuration file "
"from the command line. Format is <key>=<value>.",
short_name="D")
class Args:
"""Converts ShapNet CAD models to binary format."""
shapenet_root: str = cmd.flag("Path to ShapeNet's root directory.")
output_root: str = cmd.flag("Path to the output root directory.")
class DefaultProgramFlags(ConfigPostProcessFlags):
config_path: str = cmd_line_flags.flag("Path to config json.")
class Flags:
raw_output_path: str = cmd_line_flags.flag(
"Output checkpoint path",
default="data/raw/resnet50_weights_tf_dim_ordering_tf_kernels.h5")
output_path: str = cmd_line_flags.flag(
"Output checkpoint path", default="data/keras_resnet50_imagenet.cpt")
class ProgramArgs(pipeline.DefaultProgramFlags):
"""Trains a CoreNet model."""
recurrent_evals: bool = cmd_line_flags.flag(
"Whether to run recurrent evals.", default=True)