monai-test
/

brats_mri_generative_diffusion

Model card Files Files and versions

katielink commited on Aug 16, 2023

Commit

60f0c03

·

1 Parent(s): c35c6d7

update with new lr scheduler api

Files changed (2) hide show

configs/metadata.json +2 -1
configs/train_diffusion.json +1 -1

configs/metadata.json CHANGED Viewed

@@ -1,7 +1,8 @@
 {
     "schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_generator_ldm_20230507.json",
-    "version": "1.0.3",
     "changelog": {
         "1.0.3": "update required packages",
         "1.0.2": "unify dataset dir in different configs",
         "1.0.1": "update dependency, update trained model weights",

 {
     "schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_generator_ldm_20230507.json",
+    "version": "1.0.4",
     "changelog": {
+        "1.0.4": "update with new lr scheduler api",
         "1.0.3": "update required packages",
         "1.0.2": "unify dataset dir in different configs",
         "1.0.1": "update dependency, update trained model weights",

configs/train_diffusion.json CHANGED Viewed

@@ -59,7 +59,7 @@
         "_requires_": [
             "@load_autoencoder"
         ],
-        "beta_schedule": "scaled_linear",
         "num_train_timesteps": 1000,
         "beta_start": 0.0015,
         "beta_end": 0.0195

         "_requires_": [
             "@load_autoencoder"
         ],
+        "schedule": "scaled_linear_beta",
         "num_train_timesteps": 1000,
         "beta_start": 0.0015,
         "beta_end": 0.0195