How to get rid of this:
AttributeError: 'KerasTensor' object has no attribute 'node'
Was working fine with TF 2.4.1 but giving problem with TF 2.7.0
Script
types_inp = Input(shape=(self.len_of_types, ), dtype='int32') 
.......
out = Dense(self.num_ways, activation='softmax')(prob)
model = keras.Model(inputs=[types_inp,..],outputs=out]


Error:
AttributeError: 'KerasTensor' object has no attribute 'node'