Clean logging, Refactor

This commit is contained in:
Cadene
2024-02-29 23:13:06 +00:00
parent cb7b375526
commit 0b9027f05e
9 changed files with 229 additions and 131 deletions

View File

@@ -59,6 +59,7 @@ policy:
use_ema: true
lr_scheduler: cosine
lr_warmup_steps: 500
grad_clip_norm: 0
noise_scheduler:
# _target_: diffusers.schedulers.scheduling_ddpm.DDPMScheduler