renaming existing unet to UNet2D

qim3d/ml/models/__init__.py

-from ._unet import UNet, Hyperparameters
+from ._unet import UNet2D, Hyperparameters

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.