def initialize()

in activemri/experimental/cvpr19_models/options/base_options.py [0:0]

• 54 lines of code
• 1 McCabe index (conditional complexity)

    def initialize(self, parser):
        parser.add_argument(
            "--dataset_dir", required=True, help="Path to fastmri dataset."
        )
        parser.add_argument(
            "--dataroot",
            required=True,
            help="Path to images (should have subfolders trainA, trainB, valA, valB, etc)",
        )
        parser.add_argument(
            "--batchSize", type=int, default=1, help="Input batch size."
        )
        parser.add_argument(
            "--gpu_ids",
            type=str,
            default="0",
            help="GPU IDs: e.g. 0  0,1,2, 0,2. use -1 for CPU.",
        )
        parser.add_argument(
            "--name",
            type=str,
            default="experiment_name",
            help="Name of the experiment. It determines the sub folder where results are stored.",
        )
        parser.add_argument(
            "--nThreads", default=4, type=int, help="Number of threads for data loader."
        )
        parser.add_argument(
            "--checkpoints_dir",
            type=str,
            default="./checkpoints",
            help="Root directory to save results and model checkpoints.",
        )
        parser.add_argument(
            "--init_type",
            type=str,
            choices=["normal", "xavier", "kaiming", "orthogonal"],
            default="normal",
            help="Network weights initialization type.",
        )

        parser.add_argument(
            "--num_volumes_train",
            type=int,
            default=None,
            help="Number of MRI volumes to use for training.",
        )
        parser.add_argument(
            "--num_volumes_val",
            type=int,
            default=None,
            help="Number of MRI volumes to use for validation.",
        )

        self.initialized = True
        return parser