From 87699bff93f7c1a742ac4c0e7fd6aeceaa16b466 Mon Sep 17 00:00:00 2001 From: ganeshkumarka Date: Tue, 8 Oct 2024 22:32:13 +0530 Subject: [PATCH] Fix typo in train.py: Corrected UNe t_baseline to UNet_baseline --- Diffusion_UKAN/Diffusion/Train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Diffusion_UKAN/Diffusion/Train.py b/Diffusion_UKAN/Diffusion/Train.py index d112135..57b7460 100644 --- a/Diffusion_UKAN/Diffusion/Train.py +++ b/Diffusion_UKAN/Diffusion/Train.py @@ -9,7 +9,7 @@ # from torchvision.datasets import CIFAR10 from torchvision.utils import save_image from Diffusion import GaussianDiffusionSampler, GaussianDiffusionTrainer -from Diffusion.UNet import UNet, UNe t_Baseline +from Diffusion.UNet import UNet, UNet_Baseline from Diffusion.Model_ConvKan import UNet_ConvKan from Diffusion.Model_UMLP import UMLP from Diffusion.Model_UKAN_Hybrid import UKan_Hybrid