INNER CODE UNIT · Python

clean_cfg

KaiyangZhou/Dassl.pytorch · dassl/config/__init__.py:8

def clean_cfg(cfg, trainer):
    """Remove unused trainers (configs).

    Aim: Only show relevant information when calling print(cfg).

    Args:
        cfg (_C): cfg instance.
        trainer (str): trainer name.
    """
    keys = list(cfg.TRAINER.keys())
    for key in keys:
        if key == "NAME" or key == trainer.upper():
            continue
        cfg.TRAINER.pop(key, None)

View source record →