- sorry not sorry

inferno-pytorch · Dec 19, 2018 · b96bf92 · b96bf92
1 parent 3a226f6
commit b96bf92
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 1 deletion.
diff --git a/inferno/extensions/__init__.py b/inferno/extensions/__init__.py
@@ -4,5 +4,9 @@
 from . import layers
 from . import metrics
 from . import optimizers
+from . import models
+# Backward support
+from . import models as model
 
-__all__ = ['containers', 'criteria', 'initializers', 'layers', 'metrics', 'optimizers']
+__all__ = ['containers', 'criteria', 'initializers', 'layers', 'metrics', 'optimizers',
+           'models', 'model']
diff --git a/inferno/extensions/model/__init__.py → inferno/extensions/models/__init__.py b/inferno/extensions/model/__init__.py → inferno/extensions/models/__init__.py
diff --git a/inferno/extensions/model/res_unet.py → inferno/extensions/models/res_unet.py b/inferno/extensions/model/res_unet.py → inferno/extensions/models/res_unet.py
diff --git a/inferno/extensions/model/unet.py → inferno/extensions/models/unet.py b/inferno/extensions/model/unet.py → inferno/extensions/models/unet.py