Open forhonourlx opened 6 years ago
def dueling_dqn(input_shape, action_size, learning_rate): ... state_value = Lambda(lambda s: K.expand_dims(s[:, 0], dim=-1), output_shape=(action_size,))(state_value) ...
same issue
def dueling_dqn(input_shape, action_size, learning_rate): ... state_value = Lambda(lambda s: K.expand_dims(s[:, 0], dim=-1), output_shape=(action_size,))(state_value) ...