diff --git a/model/networks/__init__.py b/model/networks/__init__.py
index 9278aee..c6e6a30 100644
--- a/model/networks/__init__.py
+++ b/model/networks/__init__.py
@@ -5,3 +5,11 @@
     @Author : chairc
     @Site   : https://github.com/chairc
 """
+# Diffusion model network
+from .base import BaseNet
+from .cspdarkunet import CSPDarkUnet
+from .unet import UNet
+from .unetv2 import UNetV2
+
+# Super resolution network
+from .sr.srv1 import SRv1