katielink commited on
Commit
1d2c887
1 Parent(s): 2462f96

update with new lr scheduler api in inference

Browse files
configs/inference.json CHANGED
@@ -83,7 +83,7 @@
83
  "num_train_timesteps": 1000,
84
  "beta_start": 0.0015,
85
  "beta_end": 0.0195,
86
- "beta_schedule": "scaled_linear",
87
  "clip_sample": false
88
  },
89
  "noise": "$torch.randn([1]+@latent_shape).to(@device)",
 
83
  "num_train_timesteps": 1000,
84
  "beta_start": 0.0015,
85
  "beta_end": 0.0195,
86
+ "schedule": "scaled_linear_beta",
87
  "clip_sample": false
88
  },
89
  "noise": "$torch.randn([1]+@latent_shape).to(@device)",
configs/metadata.json CHANGED
@@ -1,7 +1,8 @@
1
  {
2
  "schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_generator_ldm_20230507.json",
3
- "version": "1.0.5",
4
  "changelog": {
 
5
  "1.0.5": "fix the wrong GPU index issue of multi-node",
6
  "1.0.4": "update with new lr scheduler api",
7
  "1.0.3": "update required packages",
 
1
  {
2
  "schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_generator_ldm_20230507.json",
3
+ "version": "1.0.6",
4
  "changelog": {
5
+ "1.0.6": "update with new lr scheduler api in inference",
6
  "1.0.5": "fix the wrong GPU index issue of multi-node",
7
  "1.0.4": "update with new lr scheduler api",
8
  "1.0.3": "update required packages",
configs/train_diffusion.json CHANGED
@@ -65,15 +65,15 @@
65
  "beta_start": 0.0015,
66
  "beta_end": 0.0195
67
  },
68
- "inferer": {
69
- "_target_": "generative.inferers.LatentDiffusionInferer",
70
- "scheduler": "@noise_scheduler",
71
- "scale_factor": "@scale_factor"
72
- },
73
  "loss": {
74
  "_target_": "torch.nn.MSELoss"
75
  },
76
  "train": {
 
 
 
 
 
77
  "crop_transforms": [
78
  {
79
  "_target_": "DivisiblePadd",
@@ -157,7 +157,7 @@
157
  "optimizer": "@optimizer",
158
  "loss_function": "@loss",
159
  "latent_shape": "@latent_shape",
160
- "inferer": "@inferer",
161
  "key_train_metric": "$None",
162
  "train_handlers": "@train#handlers"
163
  }
 
65
  "beta_start": 0.0015,
66
  "beta_end": 0.0195
67
  },
 
 
 
 
 
68
  "loss": {
69
  "_target_": "torch.nn.MSELoss"
70
  },
71
  "train": {
72
+ "inferer": {
73
+ "_target_": "generative.inferers.LatentDiffusionInferer",
74
+ "scheduler": "@noise_scheduler",
75
+ "scale_factor": "@scale_factor"
76
+ },
77
  "crop_transforms": [
78
  {
79
  "_target_": "DivisiblePadd",
 
157
  "optimizer": "@optimizer",
158
  "loss_function": "@loss",
159
  "latent_shape": "@latent_shape",
160
+ "inferer": "@train#inferer",
161
  "key_train_metric": "$None",
162
  "train_handlers": "@train#handlers"
163
  }