def _build_optimizers(self):
   opts = self._optimizers
   if not opts:
     raise ValueError('No optimizers provided.')
   if len(opts) == 1:
     opts = tuple([opts[0]] * 2)
   self._p_optimizer = utils.get_optimizer(opts[0][0])(lr=opts[0][1])
   self._ae_optimizer = utils.get_optimizer(opts[1][0])(lr=opts[1][1])
Esempio n. 2
0
 def _build_optimizers(self):
     opts = self._optimizers
     if len(opts) == 1:
         opts = tuple([opts[0]] * 3)
     elif len(opts) < 3:
         raise ValueError('Bad optimizers %s.' % opts)
     self._q_optimizer = utils.get_optimizer(opts[0][0])(lr=opts[0][1])
     self._p_optimizer = utils.get_optimizer(opts[1][0])(lr=opts[1][1])
     self._a_optimizer = utils.get_optimizer(opts[2][0])(lr=opts[2][1])
     if len(self._weight_decays) == 1:
         self._weight_decays = tuple([self._weight_decays[0]] * 3)
Esempio n. 3
0
 def _build_optimizers(self):
   opt = self._optimizers[0]
   opt_fn = utils.get_optimizer(opt[0])
   self._optimizer = opt_fn(lr=opt[1])