fix unet config & instiantion

config/model/image/tinyunetx.yaml

@@ -0,0 +1,22 @@
+_target_: nimrod.models.unet.TinyUnetX
+_partial_: true
+
+nnet:
+  _target_: nimrod.models.unet.TinyUnet
+  n_features: [3, 32, 64, 128, 256, 512, 1024]  # channel/feature expansion
+  pre_activation: true
+  # activation:
+  #   _target_: torch.nn.ReLU
+  #   _partial_: true
+  # leaky: 0.0
+
+  activation:
+    _target_: torch.nn.LeakyReLU
+    _partial_: true
+    negative_slope: 0.1
+
+
+  weight_initialization: true
+
+# optimizer
+# schedule

nbs/models.unet.ipynb

@@ -20,9 +20,29 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": null,
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Seed set to 42\n",
+      "/user/s/slegroux/miniconda3/envs/nimrod/lib/python3.11/site-packages/torchvision/transforms/v2/_deprecated.py:42: UserWarning: The transform `ToTensor()` is deprecated and will be removed in a future release. Instead, please use `v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)])`.Output is equivalent up to float precision.\n",
+      "  warnings.warn(\n",
+      "Seed set to 42\n",
+      "Seed set to 42\n"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "The autoreload extension is already loaded. To reload it, use:\n",
+      "  %reload_ext autoreload\n"
+     ]
+    }
+   ],
    "source": [
     "#| hide\n",
     "%load_ext autoreload\n",
@@ -32,7 +52,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 13,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -60,7 +80,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": null,
    "metadata": {},
    "outputs": [
     {
@@ -86,7 +106,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -186,7 +206,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 15,
+   "execution_count": null,
    "metadata": {},
    "outputs": [
     {
@@ -320,7 +340,7 @@
     }
    ],
    "source": [
-    "cfg = OmegaConf.load('../config/model/image/unetx.yaml')\n",
+    "cfg = OmegaConf.load('../config/model/image/tinyunetx.yaml')\n",
     "model = instantiate(cfg.nnet)\n",
     "x = torch.randn(1, 3, 64, 64)\n",
     "model(x).shape\n"
@@ -339,21 +359,9 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3",
+   "display_name": "python3",
    "language": "python",
    "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.8"
   }
  },
  "nbformat": 4,