e886d55ac1be6c34bd8ea56dba29029153d8eb5b,pl_examples/domain_templates/semantic_segmentation.py,SegModel,add_model_specific_args,#Any#,247

Before Change

    @staticmethod
    def add_model_specific_args(parent_parser):  // pragma: no-cover
        parser = ArgumentParser(parents=[parent_parser])
        parser.add_argument("--data_path", type=str, help="path where dataset is stored")
        parser.add_argument("--batch_size", type=int, default=16, help="size of the batches")
        parser.add_argument("--lr", type=float, default=0.001, help="adam: learning rate")
        parser.add_argument("--num_layers", type=int, default=5, help="number of layers on u-net")

After Change

    @staticmethod
    def add_model_specific_args(parent_parser):  // pragma: no-cover
        parser = parent_parser.add_argument_group("SegModel")
        parser.add_argument("--data_path", type=str, help="path where dataset is stored")
        parser.add_argument("--batch_size", type=int, default=16, help="size of the batches")
        parser.add_argument("--lr", type=float, default=0.001, help="adam: learning rate")
        parser.add_argument("--num_layers", type=int, default=5, help="number of layers on u-net")

Instances