Python get_training_argの例

コード例 #1

 def training_arg_was_passed(self, args, kwargs):
   if not self.layer._expects_training_arg and self._expects_training_arg:  # pylint: disable=protected-access
     return (utils.get_training_arg(self._training_arg_index, args, kwargs)
             is not None)
   else:
     return self.layer._call_arg_was_passed(  # pylint: disable=protected-access
         'training', args, kwargs, inputs_in_args=True)

コード例 #2

ファイル: save_impl.py プロジェクト: zhry600/tensorflow

 def get_training_arg_value(self, args, kwargs):
     if not self.layer._expects_training_arg and self._expects_training_arg:  # pylint: disable=protected-access
         return utils.get_training_arg(self._training_arg_index, args,
                                       kwargs)
     else:
         return self.layer._get_call_arg_value(  # pylint: disable=protected-access
             'training',
             args,
             kwargs,
             inputs_in_args=True)