def configure_optimizers()

in torchrecipes/vision/image_generation/module/infogan.py [0:0]

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

    def configure_optimizers(self) -> Tuple[List[torch.optim.Optimizer], List[Any]]:
        lr = self.hparams["optim"].lr
        b1 = self.hparams["optim"].b1
        b2 = self.hparams["optim"].b2

        optimizer_G = torch.optim.Adam(
            self.generator.parameters(), lr=lr, betas=(b1, b2)
        )
        optimizer_D = torch.optim.Adam(
            self.discriminator.parameters(), lr=lr, betas=(b1, b2)
        )
        optimizer_info = torch.optim.Adam(
            itertools.chain(
                self.generator.parameters(), self.discriminator.parameters()
            ),
            lr=lr,
            betas=(b1, b2),
        )
        return [optimizer_G, optimizer_D, optimizer_info], []