Fixing error with keras missing legacy_get_updates_support

Building keras-contrib in a [jupyter/tensorflow-notebook](https://github.com/jupyter/docker-stacks) causes the error:

> AttributeError: module 'keras.legacy.interfaces' has no attribute 'legacy_get_updates_support'

This should fix that error.
This commit is contained in:
C Reid
2017-10-05 01:00:05 -07:00
committed by GitHub
parent a92268ffcb
commit ef035e5664
-2
View File
@@ -2,7 +2,6 @@ from __future__ import absolute_import
from keras.optimizers import Optimizer
from .. import backend as K
from keras.utils.generic_utils import get_custom_objects
from keras.legacy import interfaces
class FTML(Optimizer):
@@ -31,7 +30,6 @@ class FTML(Optimizer):
self.epsilon = epsilon
self.inital_decay = decay
@interfaces.legacy_get_updates_support
def get_updates(self, loss, params):
grads = self.get_gradients(loss, params)
self.updates = [K.update_add(self.iterations, 1)]