diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-04-12 14:33:16 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-12 14:36:07 -0700 |
| commit | 0161bb77accc64d3742098feb7f438752a83ff32 (patch) | |
| tree | 1ec8c6ceb7d1210686eaf9d6115f944a3aa6e3ea /tensorflow/contrib/kfac | |
| parent | 7c0172e0853f3262e1d85aa6bc37cf70d718cca0 (diff) | |
K-FAC: Deprecate tf.contrib.kfac.
As LayerCollections are required to instantiate KfacOptimizer and FisherEstimator, a deprecation warning is printed upon instantiating LayerCollection.
PiperOrigin-RevId: 192671370
Diffstat (limited to 'tensorflow/contrib/kfac')
| -rw-r--r-- | tensorflow/contrib/kfac/python/ops/layer_collection.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/contrib/kfac/python/ops/layer_collection.py b/tensorflow/contrib/kfac/python/ops/layer_collection.py index 411da033c3..366e2a82d5 100644 --- a/tensorflow/contrib/kfac/python/ops/layer_collection.py +++ b/tensorflow/contrib/kfac/python/ops/layer_collection.py @@ -28,6 +28,7 @@ from collections import defaultdict from collections import OrderedDict from contextlib import contextmanager from functools import partial +import warnings import math import six @@ -171,6 +172,9 @@ class LayerCollection(object): def __init__(self, graph=None, name="LayerCollection"): + warnings.warn( + "tf.contrib.kfac is deprecated and will be removed by 2018-11-01. " + "Use https://pypi.python.org/pypi/kfac instead.") self.fisher_blocks = LayerParametersDict() self.fisher_factors = OrderedDict() self._linked_parameters = dict( |