weixin_40101756的博客

一般在GAN或者类似的模型里，我们会有多个网络模型，每个网络模型都需要定义各自的优化器，如下所示： def configure_optimizers(self): lr = self.hparams.lr b1 = self.hparams.b1 b2 = self.hparams.b2 opt_g = torch.optim.Adam(self.generator.parameters(), lr=lr, betas=(b1,