def parse_common_fields()

in quant/common/parser.py [0:0]

• 17 lines of code
• 8 McCabe index (conditional complexity)

def parse_common_fields(args: Namespace, config: dict) -> None:
    """
    Populate common fields in the config with parsed args.

    Args:
        args: parsed argparse CLI args
        config: config dictionary storing final resolved args
    """
    if args.experiment_name is not None:
        config['experiment_name'] = args.experiment_name
    else:
        from datetime import datetime

        current_time = datetime.now().strftime('%b%d_%H-%M-%S')
        config_name_without_ext = Path(config['config']).stem
        config['experiment_name'] = f'{current_time}_{config_name_without_ext}'

    if 'environment' not in config or 'platform' not in config['environment']:
        config['environment'] = {'platform': 'local'}

    if args.ngpus is not None:
        config['environment']['ngpus'] = args.ngpus
    if 'ngpus' not in config['environment']:
        config['environment']['ngpus'] = 1 if torch.cuda.is_available() else 0

    config['skip_training'] = args.skip_training

    if args.init_from_checkpoint:
        config['init_from_checkpoint'] = args.init_from_checkpoint