(self)
| 519 | return log |
| 520 | |
| 521 | def configure_optimizers(self): |
| 522 | lr = self.learning_rate |
| 523 | params = list(self.model.parameters()) |
| 524 | if self.learn_logvar: |
| 525 | params = params + [self.logvar] |
| 526 | opt = torch.optim.AdamW(params, lr=lr) |
| 527 | return opt |
| 528 | |
| 529 | |
| 530 | class LatentDiffusion(DDPM): |
nothing calls this directly
no outgoing calls
no test coverage detected