Ejemplos de get_optim en Python

Ejemplo n.º 1

Archivo: bot_model.py Proyecto: lufeng22/OT-ICNN

 def define_d(self, config):
     self.phi, self.psi = networks.get_d(config), networks.get_d(config)
     self.d_optimizer = networks.get_optim(
         itertools.chain(list(self.phi.parameters()),
                         list(self.psi.parameters())), config.d_lr, config)

Ejemplo n.º 2

Archivo: base_model.py Proyecto: lufeng22/OT-ICNN

 def define_g(self, config):
     self.g = networks.get_g(config)
     self.g_optimizer = networks.get_optim(self.g.parameters(), config.g_lr, config)

Ejemplo n.º 3

 def define_d(self, config):
     self.phi = networks.get_d(config)
     self.d_optimizer = networks.get_optim(self.phi.parameters(),
                                           config.d_lr, config)