コード例 #1
0
ファイル: hyper_gradients.py プロジェクト: habibrk/FAR-HO
 def iteration(self):
     return utils.flatten_list(
         [opt_dict.iteration for opt_dict in sorted(self._optimizer_dicts)])
コード例 #2
0
ファイル: hyper_gradients.py プロジェクト: codealphago/FAR-HO
 def iteration(self):
     return utils.flatten_list([opt_dict.iteration for opt_dict in sorted(self._optimizer_dicts)])
コード例 #3
0
ファイル: hyper_gradients.py プロジェクト: habibrk/FAR-HO
 def _state_feed_dict_generator(self, history):
     state = utils.flatten_list(
         [opt_dict.state for opt_dict in sorted(self._optimizer_dicts)])
     for t, his in enumerate(history):
         yield t, {v: his[k] for k, v in enumerate(state)}
コード例 #4
0
ファイル: hyper_gradients.py プロジェクト: codealphago/FAR-HO
 def _state_feed_dict_generator(self, history):
     state = utils.flatten_list([opt_dict.state for opt_dict in sorted(self._optimizer_dicts)])
     for t, his in enumerate(history):
         yield t, {v: his[k] for k, v in enumerate(state)}