Я пытаюсь взять модель, обученную ML-агентами Unity, и выполнить выводы внутри Tensorflow / Keras. Есть какие-нибудь советы, как это сделать?
Я попытался загрузить файл замороженного графа в граф TF, а затем извлечь тензор ввода / вывода. Я предполагаю, что vector_observation - это операция ввода, а действие - это операция вывода.
Но он выдал мне ошибку о action_mask при его запуске.
InvalidArgumentError: You must feed a value for placeholder tensor 'prefix/action_masks' with dtype float and shape [?,6]
[[Node: prefix/action_masks = Placeholder[dtype=DT_FLOAT, shape=[?,6], _device="/job:localhost/replica:0/task:0/device:CPU:0"]()]]
Мои возможные выходы - это 2 ветки, например: [0,1,2]; [0,1,2] .. ... так что, я думаю, поэтому было сказано, что action_masks имеет форму 6.
Я новичок в Tensorflow, прошу прощения, если я пропустил что-то глупое. Спасибо!
Вот изображение формата нейронной сети:
Входной тензор нейронной сети
Neural Network Architecutre