X train; + config

slegroux · Feb 14, 2025 · 4f79301 · 4f79301
1 parent 3dfb603
commit 4f79301
Show file tree

Hide file tree

Showing 7 changed files with 576 additions and 98 deletions.
diff --git a/config/model/image/diffusorx.yaml b/config/model/image/diffusorx.yaml
@@ -0,0 +1,19 @@
+_target_: nimrod.models.diffusion.DiffusorX
+_partial_: true # we init optim & sched later
+
+nnet:
+  _target_: diffusers.UNet2DModel
+  block_out_channels: [32, 64, 128, 256]
+  sample_size: 32
+  in_channels: 1
+  out_channels: 1
+
+noise_scheduler:
+  _target_: diffusers.DDPMScheduler
+  num_train_timesteps: 1000
+  beta_start: 0.00085
+  beta_end: 0.012
+
+
+# optimizer
+# scheduler%                 
diff --git a/nbs/models.conv.ipynb b/nbs/models.conv.ipynb
@@ -2533,13 +2533,9 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3",
+   "display_name": "python3",
    "language": "python",
    "name": "python3"
-  },
-  "language_info": {
-   "name": "python",
-   "version": "3.11.8"
   }
  },
  "nbformat": 4,

diff --git a/nbs/models.core.ipynb b/nbs/models.core.ipynb
@@ -1341,13 +1341,9 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3",
+   "display_name": "python3",
    "language": "python",
    "name": "python3"
-  },
-  "language_info": {
-   "name": "python",
-   "version": "3.11.8"
   }
  },
  "nbformat": 4,

diff --git a/nbs/models.diffusion.ipynb b/nbs/models.diffusion.ipynb
diff --git a/nbs/models.resnet.ipynb b/nbs/models.resnet.ipynb
@@ -698,13 +698,9 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3",
+   "display_name": "python3",
    "language": "python",
    "name": "python3"
-  },
-  "language_info": {
-   "name": "python",
-   "version": "3.11.8"
   }
  },
  "nbformat": 4,

diff --git a/nbs/models.unet.ipynb b/nbs/models.unet.ipynb
@@ -359,13 +359,9 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3",
+   "display_name": "python3",
    "language": "python",
    "name": "python3"
-  },
-  "language_info": {
-   "name": "python",
-   "version": "3.11.8"
   }
  },
  "nbformat": 4,

diff --git a/tutorials/diffusion.ipynb b/tutorials/diffusion.ipynb