af02ac9f01bf36b0c278d85636e91ff6b483efb4,tests/test_hooks.py,,_build_demo_runner,#,183

Before Change

def _build_demo_runner():
    model = nn.Linear(2, 1)
    work_dir = osp.join(osp.dirname(osp.abspath(__file__)), "data")
    optimizer = torch.optim.SGD(model.parameters(), lr=0.02, momentum=0.95)

    log_config = dict(
        interval=1, hooks=[

After Change

        work_dir=tmp_dir,
        batch_processor=lambda model, x, **kwargs: {"loss": model(x) - 0},
        optimizer=optimizer,
        logger=logging.getLogger())

    runner.register_logger_hooks(log_config)
    return runner

Instances