Python add_module_changing_check Examples

Example #1

File: test_disable_enable.py Project: pengyan510/torcheck

def test_disable(unchanging_model_optimizer, unchanging_model, dataloader,
                 run_training):
    torcheck.register(unchanging_model_optimizer)
    torcheck.add_module_changing_check(unchanging_model,
                                       module_name="NeuralNet")
    torcheck.disable()
    run_training(unchanging_model, dataloader, unchanging_model_optimizer)

Example #2

File: test_param_check.py Project: pengyan510/torcheck

def test_module_changing_check_with_unchanging_model(
    unchanging_model_optimizer, unchanging_model, dataloader, run_training
):
    torcheck.register(unchanging_model_optimizer)
    torcheck.add_module_changing_check(unchanging_model, module_name="NeuralNet")
    with pytest.raises(
        RuntimeError,
        match=(
            r"Module NeuralNet's fc1\.weight should change\.\n"
            r".*fc1.bias should change"
        ),
    ):
        run_training(unchanging_model, dataloader, unchanging_model_optimizer)

Example #3

def test_verbose_on(
    unchanging_model_optimizer, unchanging_model, dataloader, run_training
):
    torcheck.verbose_on()
    torcheck.register(unchanging_model_optimizer)
    torcheck.add_module_changing_check(unchanging_model, module_name="NeuralNet")
    with pytest.raises(
        RuntimeError,
        match=(
            r"Module NeuralNet's fc1\.weight should change\.\n"
            r"The tensor is:(.|\n)*"
            r"fc1\.bias should change\.\n"
            r"The tensor is:(.|\n)*"
        ),
    ):
        run_training(unchanging_model, dataloader, unchanging_model_optimizer)

Example #4

File: test_param_check.py Project: pengyan510/torcheck

def test_module_changing_check_with_changing_model(
    changing_model_optimizer, changing_model, dataloader, run_training
):
    torcheck.register(changing_model_optimizer)
    torcheck.add_module_changing_check(changing_model, module_name="NeuralNet")
    run_training(changing_model, dataloader, changing_model_optimizer)