experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

remove optimizer checking.

Browse Source
This commit is contained in:
apantykhin 2018-06-06 19:55:46 +04:00
parent 8dc3b3c453
commit 558cbd9fc8
1 changed files with 0 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/contrib/gan/python/estimator/python/head_impl.py
View File

@ -25,7 +25,6 @@ from tensorflow.contrib.gan.python import train as tfgan_train
from tensorflow.python.estimator import model_fn as model_fn_lib from tensorflow.python.estimator import model_fn as model_fn_lib
from tensorflow.python.estimator.canned import head from tensorflow.python.estimator.canned import head
from tensorflow.python.framework import ops from tensorflow.python.framework import ops
from tensorflow.python.training import optimizer
__all__ = [ __all__ = [
'GANHead', 'GANHead',
@ -96,10 +95,6 @@ class GANHead(head._Head): # pylint: disable=protected-access
raise TypeError('generator_loss_fn must be callable.') raise TypeError('generator_loss_fn must be callable.')
if not callable(discriminator_loss_fn): if not callable(discriminator_loss_fn):
raise TypeError('discriminator_loss_fn must be callable.') raise TypeError('discriminator_loss_fn must be callable.')
if not isinstance(generator_optimizer, optimizer.Optimizer):
raise TypeError('generator_optimizer must be Optimizer.')
if not isinstance(discriminator_optimizer, optimizer.Optimizer):
raise TypeError('discriminator_optimizer must be Optimizer.')
if not use_loss_summaries in [True, False, None]: if not use_loss_summaries in [True, False, None]:
raise ValueError('use_loss_summaries must be True, False or None.') raise ValueError('use_loss_summaries must be True, False or None.')
if get_hooks_fn is not None and not callable(get_hooks_fn): if get_hooks_fn is not None and not callable(get_hooks_fn):