(main_checkpoint, step, rate)
| 338 | |
| 339 | |
| 340 | def find_ema_checkpoint(main_checkpoint, step, rate): |
| 341 | if main_checkpoint is None: |
| 342 | return None |
| 343 | filename = f"ema_{rate}_{(step):06d}.pt" |
| 344 | path = bf.join(bf.dirname(main_checkpoint), filename) |
| 345 | if bf.exists(path): |
| 346 | return path |
| 347 | return None |
| 348 | |
| 349 | |
| 350 | def log_loss_dict(diffusion, ts, losses): |
no outgoing calls
no test coverage detected