katielink commited on
Commit
993902a
1 Parent(s): 7a11f9f

update with new lr scheduler api in inference

Browse files
configs/inference.json CHANGED
@@ -78,7 +78,7 @@
78
  "num_train_timesteps": 1000,
79
  "beta_start": 0.0015,
80
  "beta_end": 0.0195,
81
- "beta_schedule": "scaled_linear",
82
  "clip_sample": false
83
  },
84
  "noise": "$torch.randn([1]+@latent_shape).to(@device)",
 
78
  "num_train_timesteps": 1000,
79
  "beta_start": 0.0015,
80
  "beta_end": 0.0195,
81
+ "schedule": "scaled_linear_beta",
82
  "clip_sample": false
83
  },
84
  "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
@@ -64,15 +64,15 @@
64
  "beta_start": 0.0015,
65
  "beta_end": 0.0195
66
  },
67
- "inferer": {
68
- "_target_": "generative.inferers.LatentDiffusionInferer",
69
- "scheduler": "@noise_scheduler",
70
- "scale_factor": "@scale_factor"
71
- },
72
  "loss": {
73
  "_target_": "torch.nn.MSELoss"
74
  },
75
  "train": {
 
 
 
 
 
76
  "crop_transforms": [
77
  {
78
  "_target_": "CenterSpatialCropd",
@@ -140,7 +140,7 @@
140
  "optimizer": "@optimizer",
141
  "loss_function": "@loss",
142
  "latent_shape": "@latent_shape",
143
- "inferer": "@inferer",
144
  "key_train_metric": "$None",
145
  "train_handlers": "@train#handlers"
146
  }
 
64
  "beta_start": 0.0015,
65
  "beta_end": 0.0195
66
  },
 
 
 
 
 
67
  "loss": {
68
  "_target_": "torch.nn.MSELoss"
69
  },
70
  "train": {
71
+ "inferer": {
72
+ "_target_": "generative.inferers.LatentDiffusionInferer",
73
+ "scheduler": "@noise_scheduler",
74
+ "scale_factor": "@scale_factor"
75
+ },
76
  "crop_transforms": [
77
  {
78
  "_target_": "CenterSpatialCropd",
 
140
  "optimizer": "@optimizer",
141
  "loss_function": "@loss",
142
  "latent_shape": "@latent_shape",
143
+ "inferer": "@train#inferer",
144
  "key_train_metric": "$None",
145
  "train_handlers": "@train#handlers"
146
  }