def configure_optimizers()

in projects/variational_image_compression/lightning/_prior_autoencoder.py [0:0]

• 22 lines of code
• 1 McCabe index (conditional complexity)

    def configure_optimizers(self):
        parameters, bottleneck_parameters = self.network.group_parameters()

        optimizer = Adam(
            parameters.values(),
            self.hparams.optimizer_lr,
        )

        bottleneck_optimizer = Adam(
            bottleneck_parameters.values(),
            self.hparams.bottleneck_optimizer_lr,
        )

        return (
            {
                "lr_scheduler": {
                    "scheduler": ReduceLROnPlateau(optimizer, "min"),
                    "monitor": "val_rate_distortion",
                },
                "optimizer": optimizer,
            },
            {
                "optimizer": bottleneck_optimizer,
            },
        )