LSTM печать градиентов керас - PullRequest
Новая
       9

LSTM печать градиентов керас

0 голосов
/ 03 октября 2018

Я пытаюсь напечатать градиенты, и у меня пока есть это.

my_model=load_model('word2vec_lstm.h5')

weights = my_model.weights # weight tensors

gradients = my_model.optimizer.get_gradients(my_model.total_loss, weights) # gradient tensors

input_tensors = [my_model.inputs[0], # input data
                 my_model.sample_weights[0], # sample weights
                 my_model.targets[0], # labels
                 K.learning_phase(), # train or test mode
]

get_gradients = K.function(inputs=input_tensors, outputs=gradients)

inputs = [[trainDataVecs], # X input data
          [1], # sample weights
          [Y_train], # y labels
          0.005 # learning phase in TEST mode
]

print [a for a in zip(weights, get_gradients(inputs))]

Однако, это дает мне следующую ошибку InvalidArgumentError: transpose expects a vector of size 4. But input(1) is a vector of size 3

...