def METHOD_NAME(self, grad_acc: int):
    result_dir = self.run(
        self.FIXTURES_ROOT / "integrations" / "torch" / "train.jsonnet",
        include_package=[
            "test_fixtures.integrations.common",
            "test_fixtures.integrations.torch",
        ],
        overrides=json.dumps(
            {
                "steps.train.train_steps": None,
                "steps.train.train_epochs": 2,
                "steps.train.validate_every": None,
                "steps.train.grad_accum": grad_acc,
            }
        ),
    )
    assert (result_dir / "train" / "data.pt").is_file()
    # Make sure we trained for the right number of steps.
    expected_steps = 16 // grad_acc
    latest = result_dir / "train" / "work" / "checkpoint_state_latest"
    assert latest.is_symlink()
    last_step = result_dir / "train" / "work" / f"checkpoint_state_step{expected_steps}"
    assert last_step.is_dir()
    assert latest.samefile(last_step)