test_padis_slurm_training_tasks.py

Source: tests/experiments/test_padis_slurm_training_tasks.py

Test padis slurm training tasks behaviour.

LION_ROOT

COMMON

SUBMIT_ALL_TRAINING

TRAINING_ARRAY

LEGACY_LIDC_256

PNP_TRAINING

test_a100_training_tasks_cover_reconstruction_model_families()

Verify that a100 training tasks cover reconstruction model families.

test_a100_training_tasks_use_expected_engines_for_special_priors()

Verify that a100 training tasks use expected engines for special priors.

test_a100_p96_batch_size_is_capped_at_96()

Verify that a100 p96 batch size is capped at 96.

test_a100_training_tasks_produce_matrix_checkpoint_filenames()

Verify that a100 training tasks produce matrix checkpoint filenames.

test_pnp_training_default_final_checkpoint_matches_reconstruction_matrix()

Verify that pnp training default final checkpoint matches reconstruction matrix.

test_slurm_training_defaults_to_lion_dev_with_padis_dev_fallback()

Verify that slurm training defaults to lion dev with padis dev fallback.

test_a100_training_array_keeps_wandb_artifacts_enabled_by_default()

Verify that a100 training array keeps wandb artifacts enabled by default.

test_a100_training_array_runs_intensive_validation_for_final_six_hours()

Verify that a100 training array runs intensive validation for final six hours.

test_all_training_submitter_launches_diffusion_array_and_pnp_denoiser(tmp_path)

Verify that all training submitter launches diffusion array and pnp denoiser.

test_all_training_submitter_is_executable()

Verify that all training submitter is executable.

test_all_training_submitter_can_skip_pnp_training(tmp_path)

Verify that all training submitter can skip pnp training.