def set_model_requires_grad(self, name, requires_grad):
        """Sets the `requires_grad` configuration for a particular model."""
        for param in self.models[name].parameters():
            param.requires_grad = requires_grad
Sets the `requires_grad` configuration for a particular model.
runners/base_runner.py