update with new lr scheduler api
Browse files- configs/metadata.json +2 -1
- configs/train_diffusion.json +1 -1
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.
|
4 |
"changelog": {
|
|
|
5 |
"1.0.3": "update required packages",
|
6 |
"1.0.2": "unify dataset dir in different configs",
|
7 |
"1.0.1": "update dependency, update trained model weights",
|
|
|
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.4",
|
4 |
"changelog": {
|
5 |
+
"1.0.4": "update with new lr scheduler api",
|
6 |
"1.0.3": "update required packages",
|
7 |
"1.0.2": "unify dataset dir in different configs",
|
8 |
"1.0.1": "update dependency, update trained model weights",
|
configs/train_diffusion.json
CHANGED
@@ -59,7 +59,7 @@
|
|
59 |
"_requires_": [
|
60 |
"@load_autoencoder"
|
61 |
],
|
62 |
-
"
|
63 |
"num_train_timesteps": 1000,
|
64 |
"beta_start": 0.0015,
|
65 |
"beta_end": 0.0195
|
|
|
59 |
"_requires_": [
|
60 |
"@load_autoencoder"
|
61 |
],
|
62 |
+
"schedule": "scaled_linear_beta",
|
63 |
"num_train_timesteps": 1000,
|
64 |
"beta_start": 0.0015,
|
65 |
"beta_end": 0.0195
|