projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 36b3c94) | patch
Fix the issue where gpu_option is not respected for keras estimator.
authorYifei Feng <yifeif@google.com>
Mon, 5 Mar 2018 21:47:30 +0000 (13:47 -0800)
committerTensorFlower Gardener <gardener@tensorflow.org>
Mon, 5 Mar 2018 21:52:51 +0000 (13:52 -0800)
commit355fb5e14b325a1d106c4046f478da4bda350205
tree5417ca00c0dcc5188d6695efc9788f87c3439b74tree | snapshot
parent36b3c94a99704c8e1973ae5c043aec4870ae84ffcommit | diff
Fix the issue where gpu_option is not respected for keras estimator.
Set keras backend session with the given config before any get_session call creates a new session.
Fix #14776.

PiperOrigin-RevId: 187916300
tensorflow/python/keras/_impl/keras/estimator.py diff | blob | history
tensorflow/python/keras/_impl/keras/estimator_test.py diff | blob | history
Domain: Machine Learning / ML Framework;