From a5d5f30fbd9c6c7c78834072401932c84bddaf14 Mon Sep 17 00:00:00 2001 From: gdamms Date: Thu, 5 Feb 2026 15:33:20 +0100 Subject: trying to improve whole project --- src/__init__.py | 50 ++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 50 insertions(+) create mode 100644 src/__init__.py (limited to 'src/__init__.py') diff --git a/src/__init__.py b/src/__init__.py new file mode 100644 index 0000000..220bb0e --- /dev/null +++ b/src/__init__.py @@ -0,0 +1,50 @@ +""" +MNIST Diffusion Model package. +""" + +from .config import ( + DEVICE, + IMG_SIZE, + NB_CHANNEL, + NB_LABEL, + DIFFU_STEPS, + ALPHA, + ALPHA_BAR, + BETA, + EPOCHS, + BATCH_SIZE, + LEARNING_RATE, + CHECKPOINT_DIR, + PLOTS_DIR, +) + +from .diffusion import ( + q_xt_xt_1, + q_xt_x0, + p_xt_1_xt, + p_xt_1_xt_x0_pred, + forward_diffusion, +) + +from .dataloader import ( + get_mnist_dataset, + get_diffusion_dataloader, + get_autoencoder_dataloader, + DiffusionDataset, + DiffusionDatasetX0, + AutoencoderDataset, +) + +from .utils import ( + tensor_to_image, + tensor_to_images, + save_checkpoint, + load_checkpoint, + save_plot, +) + +from .metrics import ( + fid, + kl_divergence, + jsd, +) -- cgit v1.3.1