HReynaud/EchoDiffusionDemo

1unets:2  unet1:3    dim: 644    num_resnet_blocks: 25    dim_mults:6    - 17    - 28    - 49    max_text_len: 110    layer_attns: false11    layer_cross_attns:12    - false13    - false14    - true15    cond_images_channels: 316    attend_at_middle: false17imagen:18  elucidated: True19  condition_on_text: true20  image_sizes:21  - 11222  text_embed_dim: 123  num_sample_steps: 6424  temporal_downsample_factor:25  - 126  lowres_sample_noise_level: 0.227  sigma_min: 0.00228  sigma_max: 8029  sigma_data: 0.2530  rho: 731  P_mean: -1.232  P_std: 1.233  S_churn: 16034  S_tmin: 0.0535  S_tmax: 5036  S_noise: 1.00337trainer:38  split_batches: false39  lr: 0.000540  dl_tuple_output_keywords_names:41  - images42  - text_embeds43  - cond_images44dataset:45  data_path: /46  deactivate_cache: false47  fps: 848  duration: 2.049  grayscale: false50dataloader:51  batch_size: 852  num_workers: 853wandb:54  project: ""55  entity: ""56checkpoint:57  path: /58  batch_size: 459  cond_scale: 5.060  save_every_x_it: 500061