Esempi in Python per get_optim

Esempio n. 1

File: bot_model.py Progetto: 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)

Esempio n. 2

File: base_model.py Progetto: 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)

Esempio 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)