check_padis_public_recon_alignment.py π§ΒΆ
Source: scripts/dev/check_padis_public_recon_alignment.py
Warning
This file has not yet received a complete narrative and docstring audit. Its public source-level API is listed automatically below.
Focused checks for LION PaDIS public-repo reconstruction compatibility.
- PROJECT_ROOT
- PADIS_REPO
- class PublicStyleDenoiser
Analytic denoiser exposing the public PaDIS model interface.
- PublicStyleDenoiser.forward(self, x, sigma, x_pos=None, class_labels=None)
No docstring is available.
- PublicStyleDenoiser.round_sigma(self, sigma)
No docstring is available.
- class RawEDMModel(*, pad_width=2, patch_size=5)
Wrap an analytic denoiser with raw EDM preconditioning semantics.
- RawEDMModel.forward(self, model_input, c_noise)
No docstring is available.
- class IdentityOperator(shape)
Small identity forward model used to isolate sampler mechanics.
- IdentityOperator.forward(self, x)
No docstring is available.
- IdentityOperator.A(self, x)
No docstring is available.
- IdentityOperator.adjoint(self, y)
No docstring is available.
- IdentityOperator.AT(self, y)
No docstring is available.
- IdentityOperator.inverse(self, y)
No docstring is available.
- IdentityOperator.Adagger(self, y)
No docstring is available.
- IdentityOperator.domain_shape(self)
No docstring is available.
- IdentityOperator.range_shape(self)
No docstring is available.
- make_lion_reconstructor(*, image_size=8, pad_width=2, patch_size=5)
No docstring is available.
- full_position_grid(height, width, device)
No docstring is available.
- assert_close(name, got, expected, atol=2e-05, rtol=2e-05)
No docstring is available.
- test_noise_schedule()
No docstring is available.
- test_patch_layout_and_denoising()
No docstring is available.
- test_dps_loop_identity_operator()
No docstring is available.
- main()
Run deterministic public/LION sampler-alignment assertions.