| if_resume_training: false | |
| device: cuda | |
| load: false | |
| load_path: '' | |
| dataset_main_path: ./datasets | |
| dataset_annotation_path: ./datasets/annotations | |
| dataset_precomputed_path: ./datasets/processed | |
| output_log_path: ./trained_models/hmp/amass/diffusion | |
| pretrained_autoencoder_path: ./trained_models/hmp/amass/autoencoder/checkpoints/cvpr_release.pt | |
| lr: 0.001 | |
| diffusion_objective: pred_x0 | |
| weight_decay: 0.0 | |
| if_use_ema: true | |
| step_start_ema: 100 | |
| ema_power: 0.6666666666666666 | |
| ema_update_every: 10 | |
| ema_min_value: 0.0 | |
| use_lr_scheduler: true | |
| lr_scheduler_kwargs: | |
| lr_scheduler_type: ExponentialLRSchedulerWarmup | |
| warmup_duration: 75 | |
| update_every: 2 | |
| min_lr: 0.0001 | |
| gamma_decay: 0.98 | |
| diffusion_conditioning: true | |
| num_epochs: 150 | |
| num_workers: 4 | |
| batch_size: 64 | |
| batch_size_eval: 256 | |
| eval_frequency: 25 | |
| train_pick_best_sample_among_k: 50 | |
| similarity_space: input_space | |
| diffusion_activation: identity | |
| num_prob_samples: 50 | |
| diffusion_timesteps: 10 | |
| diffusion_type: NonisotropicGaussianDiffusion | |
| diffusion_loss_type: snr | |
| loss_reduction_type: l1 | |
| if_run_as_isotropic: false | |
| if_sigma_n_scale: true | |
| diffusion_covariance_type: skeleton-diffusion | |
| gamma_scheduler: cosine | |
| beta_schedule: cosine | |
| sigma_n_scale: spectral | |
| num_iter_perepoch: null | |
| seed: 63485 | |
| diffusion_arch: | |
| arch: Denoiser | |
| use_attention: true | |
| self_condition: false | |
| norm_type: none | |
| depth: 4 | |
| attn_dim_head: 32 | |
| attn_heads: 8 | |
| learn_influence: true | |
| covariance_matrix_type: adjacency | |