Проблема при попытке конвертировать .pb в .nn с помощью tensbow_to_barracuda.py

Question

Я получаю следующую ошибку

Converting C:\Users\qwerty\Documents\saved_model.pb to output.nn
Sorting model, may take a while.... Done!
IGNORED: GatherV2 unknown layer
Traceback (most recent call last):
  File "tensorflow_to_barracuda.py", line 26, in <module>
    tf2bc.convert(args.source_file, args.target_file, args.trim_unused_by_output, args)
  File "C:\Repos\ml-agents\ml-agents\mlagents\trainers\tensorflow_to_barracuda.py", line 1552, in convert
    i_model, args
  File "C:\Repos\ml-agents\ml-agents\mlagents\trainers\tensorflow_to_barracuda.py", line 1377, in process_model
    process_layer(n, o_context, args)
  File "C:\Repos\ml-agents\ml-agents\mlagents\trainers\tensorflow_to_barracuda.py", line 1220, in process_layer
    assert all_elements_equal(input_ranks)
AssertionError

Что мне нужно сделать, чтобы это исправить? или есть ли другой способ преобразовать файл .pb в файл .nn для запуска моей предварительно обученной модели в Unity Inference Engine

Answer 1

Если вы используете Tensorflow v2, вам нужна более старая Tensorflow версия. (x <2) Несмотря на наличие исправлений, они могут содержать ошибки, связанные с <code>Tensorflow v2. Удалите его и установите 1.15rc2 (например).