(main_checkpoint, step, rate)
| 283 | |
| 284 | |
| 285 | def find_ema_checkpoint(main_checkpoint, step, rate): |
| 286 | if main_checkpoint is None: |
| 287 | return None |
| 288 | filename = f"ema_{rate}_{(step):06d}.pt" |
| 289 | path = bf.join(bf.dirname(main_checkpoint), filename) |
| 290 | if bf.exists(path): |
| 291 | return path |
| 292 | return None |
| 293 | |
| 294 | |
| 295 | def log_loss_dict(diffusion, ts, losses): |
no outgoing calls
no test coverage detected