Method restore

skills/paper2code/worked/ddpm/src/utils.py:239–243

Restore original model weights (after evaluation).

(self, model: nn.Module)

                param.data.copy_(self.shadow[name])

    def restore(self, model: nn.Module):
        """Restore original model weights (after evaluation)."""
        # NOTE: This requires storing original weights separately.
        # The caller should save model.state_dict() before calling apply().
        pass