Hi,
I have a problem
please guide me
thank you
from keras.layers import Input, LSTM, concatenate, Dense, Lambda
from keras.models import Model
input_ = Input(shape=(30, 6), name='input')
input_1 = Lambda(lambda x: x[:, :15, :])(input)
input_2 = Lambda(lambda x: x[:, 15:, :])(input)
lstm1 = LSTM(256, name='lstm1')(input_1)
lstm2 = LSTM(256, name='lstm2')(input_2)
concat = concatenate([lstm1, lstm2])
output = Dense(6, activation='tanh', name='dense')(concat)
model = Model(inputs=input_, outputs=output)
What I have tried:
Hi,
I have a problem
please guide me
How to train this code with Federated Averaging (FedAvg)?