Clean logging, Refactor

2026-05-26 14:09:47 +00:00 · 2024-02-29 23:13:06 +00:00
parent cb7b375526
commit 0b9027f05e
9 changed files with 229 additions and 131 deletions
@@ -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