doc: options auto update

jolibrain · Sep 11, 2024 · d05b337 · d05b337
1 parent ed701ad
commit d05b337
Show file tree

Hide file tree

Showing 3 changed files with 64 additions and 52 deletions.
diff --git a/docs/options.md b/docs/options.md
@@ -77,7 +77,11 @@ Here are all the available options to call with `train.py`
 
 | Parameter | Type | Default | Description |
 | --- | --- | --- | --- |
+| --alg_cm_dists_mean | array | [0.485, 0.456, 0.406] | mean for DISTS perceptual loss |
+| --alg_cm_dists_std | array | [0.229, 0.224, 0.225] | std for DISTS perceptual loss |
+| --alg_cm_lambda_perceptual | float | 1.0 | weight for LPIPS and DISTS perceptual losses |
 | --alg_cm_num_steps | int | 1000000 | number of steps before reaching the fully discretized consistency model sampling schedule |
+| --alg_cm_perceptual_loss | array | [''] | optional supervised perceptual loss<br/><br/> **Values:** , LPIPS, DISTS |
 | --alg_diffusion_cond_computed_sketch_list | array | ['canny', 'hed'] | what primitives to use for random sketch |
 | --alg_diffusion_cond_embed | string |  | whether to use conditioning embeddings to the generator layers, and what type<br/><br/> **Values:** , mask, class, mask_and_class, ref |
 | --alg_diffusion_cond_embed_dim | int | 32 | nb of examples processed for inference |

diff --git a/docs/source/_static/openapi.json b/docs/source/_static/openapi.json