From b2a8a35c0d874b0b7f5408d2ae2cec2a023f44a4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Anna=20B=C3=B8gevang=20Ekner?= <s193396@dtu.dk>
Date: Tue, 4 Feb 2025 14:13:12 +0100
Subject: [PATCH] renaming existing unet to UNet2D
---
qim3d/ml/models/__init__.py | 2 +-
qim3d/ml/models/_unet.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/qim3d/ml/models/__init__.py b/qim3d/ml/models/__init__.py
index 4624be5f..65493e60 100644
--- a/qim3d/ml/models/__init__.py
+++ b/qim3d/ml/models/__init__.py
@@ -1 +1 @@
-from ._unet import UNet, Hyperparameters
+from ._unet import UNet2D, Hyperparameters
diff --git a/qim3d/ml/models/_unet.py b/qim3d/ml/models/_unet.py
index fa41c021..40006ad4 100644
--- a/qim3d/ml/models/_unet.py
+++ b/qim3d/ml/models/_unet.py
@@ -5,7 +5,7 @@ import torch.nn as nn
from qim3d.utils import log
-class UNet(nn.Module):
+class UNet2D(nn.Module):
"""
2D UNet model for QIM imaging.
--
GitLab