| wandb_group: lidm_cardiacnet | |
| output_dir: /nfs/usrhome/khmuhammad/Echodream/experiments/lidm_cardiacnet | |
| vae_path: /nfs/usrhome/khmuhammad/Echodream/models/vae | |
| globals: | |
| target_fps: 32 | |
| target_nframes: 64 | |
| outputs: | |
| - image | |
| - class_id | |
| datasets: | |
| - name: CardiacNetLatent | |
| active: true | |
| params: | |
| root: /nfs/usrhome/khmuhammad/Echodream/data/latents/cardiacnet | |
| target_fps: ${globals.target_fps} | |
| target_nframes: ${globals.target_nframes} | |
| target_resolution: 14 | |
| outputs: ${globals.outputs} | |
| unet: | |
| _class_name: UNet2DConditionModel | |
| sample_size: 14 | |
| in_channels: 4 | |
| out_channels: 4 | |
| center_input_sample: false | |
| time_embedding_type: positional | |
| freq_shift: 0 | |
| flip_sin_to_cos: true | |
| down_block_types: | |
| - AttnDownBlock2D | |
| - AttnDownBlock2D | |
| - AttnDownBlock2D | |
| - DownBlock2D | |
| up_block_types: | |
| - UpBlock2D | |
| - AttnUpBlock2D | |
| - AttnUpBlock2D | |
| - AttnUpBlock2D | |
| block_out_channels: | |
| - 128 | |
| - 256 | |
| - 256 | |
| - 512 | |
| layers_per_block: 2 | |
| mid_block_scale_factor: 1 | |
| downsample_padding: 1 | |
| dropout: 0.0 | |
| act_fn: silu | |
| cross_attention_dim: 1 | |
| norm_eps: 1.0e-05 | |
| noise_scheduler: | |
| _class_name: DDPMScheduler | |
| num_train_timesteps: 1000 | |
| beta_start: 0.0001 | |
| beta_end: 0.02 | |
| beta_schedule: linear | |
| variance_type: fixed_small | |
| clip_sample: true | |
| clip_sample_range: 4.0 | |
| prediction_type: v_prediction | |
| thresholding: false | |
| dynamic_thresholding_ratio: 0.995 | |
| sample_max_value: 1.0 | |
| timestep_spacing: leading | |
| steps_offset: 0 | |
| training_mode: diffusion | |
| train_batch_size: 256 | |
| dataloader_num_workers: 16 | |
| max_train_steps: 5000 | |
| training_conditioning_type: class_id | |
| learning_rate: 0.0003 | |
| lr_warmup_steps: 500 | |
| scale_lr: false | |
| lr_scheduler: constant | |
| use_8bit_adam: false | |
| gradient_accumulation_steps: 1 | |
| noise_offset: 0.0 | |
| drop_conditionning: 0.1 | |
| gradient_checkpointing: false | |
| use_ema: true | |
| enable_xformers_memory_efficient_attention: false | |
| allow_tf32: true | |
| adam_beta1: 0.9 | |
| adam_beta2: 0.999 | |
| adam_weight_decay: 0.01 | |
| adam_epsilon: 1.0e-08 | |
| max_grad_norm: 1.0 | |
| logging_dir: logs | |
| mixed_precision: fp16 | |
| validation_timesteps: 128 | |
| validation_fps: ${globals.target_fps} | |
| validation_frames: ${globals.target_nframes} | |
| validation_count: 4 | |
| validation_guidance: 1.0 | |
| validation_steps: 100 | |
| validation_conditioning_type: class_id | |
| report_to: wandb | |
| checkpointing_steps: 100 | |
| checkpoints_total_limit: 50 | |
| resume_from_checkpoint: null | |
| tracker_project_name: echo-dream | |
| seed: 42 | |
| text_encoder_path: openai/clip-vit-large-patch14 | |
| pretrained_model_name_or_path: openai/clip-vit-large-patch14 | |
| tokenizer_path: openai/clip-vit-large-patch14 | |
| train_text_encoder: false | |
| condition_guidance_scale: 5.0 | |
| num_validation_samples: 4 | |
| num_train_epochs: 5000 | |