Skip to content

Commit 4b749ef

Browse files
update noise scheduler api for ldm (#1417)
Fixes # . ### Description update [noise scheduler api](https://github.com/Project-MONAI/GenerativeModels/blob/b1c1a459c98a4d55570fe775fdcfad1707dae71b/generative/networks/schedulers/ddpm.py#L75) for ldm. corresponding [PR](https://github.com/Project-MONAI/model-zoo/pull/449/files) in bundle ### Checks <!--- Put an `x` in all the boxes that apply, and remove the not applicable items --> - [ ] Avoid including large-size files in the PR. - [ ] Clean up long text outputs from code cells in the notebook. - [ ] For security purposes, please check the contents and remove any sensitive info such as user names and private key. - [ ] Ensure (1) hyperlinks and markdown anchors are working (2) use relative paths for tutorial repo files (3) put figure and graphs in the `./figure` folder - [ ] Notebook runs automatically `./runner.sh -t <path to .ipynb file>` Signed-off-by: Can-Zhao <[email protected]> Co-authored-by: Yiheng Wang <[email protected]>
1 parent 5974486 commit 4b749ef

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

generative/2d_ldm/train_diffusion.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,7 @@ def main():
148148

149149
scheduler = DDPMScheduler(
150150
num_train_timesteps=args.NoiseScheduler["num_train_timesteps"],
151-
beta_schedule="scaled_linear",
151+
schedule="scaled_linear_beta",
152152
beta_start=args.NoiseScheduler["beta_start"],
153153
beta_end=args.NoiseScheduler["beta_end"],
154154
)

generative/3d_ldm/train_diffusion.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,7 @@ def main():
144144

145145
scheduler = DDPMScheduler(
146146
num_train_timesteps=args.NoiseScheduler["num_train_timesteps"],
147-
beta_schedule="scaled_linear",
147+
schedule="scaled_linear_beta",
148148
beta_start=args.NoiseScheduler["beta_start"],
149149
beta_end=args.NoiseScheduler["beta_end"],
150150
)

0 commit comments

Comments
 (0)