llouislu / tensorflow-dain Goto Github PK

Hi
When I try to execute the code, I encounter the following call error.
Thank you.
Help me.

AttributeError Traceback (most recent call last)

in ()
15 #---------------USE DAIN-----------------
16 #Adaptive Input Normalization layer
---> 17 from dain import Dain

/content/drive/MyDrive/real-------------/dain.py in ()
4
5
----> 6 class Dain(layers.LaDainyer):
7 def init(self, dim, n_features, name="dain", **kwargs):
8 # input tensor: (batch_size, rows, n_features)

AttributeError: module 'keras.api._v2.keras.layers' has no attribute 'LaDainyer'

fn = lambda elem: K.switch(K.less_equal(elem, 1.0), K.ones_like(elem), elem)
adaptive_std = K.map_fn(fn, adaptive_std)

In author's code is
adaptive_std[adaptive_std <= self.eps] = 1

which self.eps = 1e-8