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])
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)
def _build_optimizers(self): opt = self._optimizers[0] opt_fn = utils.get_optimizer(opt[0]) self._optimizer = opt_fn(lr=opt[1])