(self, hparams, prior_model=None, mean=None, std=None)
| 615 | |
| 616 | class LNNP(LightningModule): |
| 617 | def __init__(self, hparams, prior_model=None, mean=None, std=None): |
| 618 | super(LNNP, self).__init__() |
| 619 | |
| 620 | self.save_hyperparameters(hparams) |
| 621 | |
| 622 | if self.hparams.load_model: |
| 623 | self.model = load_model(self.hparams.load_model, args=self.hparams) |
| 624 | else: |
| 625 | self.model = create_model(self.hparams, prior_model, mean, std) |
| 626 | |
| 627 | self._reset_losses_dict() |
| 628 | self._reset_ema_dict() |
| 629 | self._reset_inference_results() |
| 630 | |
| 631 | def configure_optimizers(self): |
| 632 | optimizer = AdamW( |
no test coverage detected