Проблема с загрузкой модели tf.keras (Yolov3)

Question

Я пытаюсь загрузить сохраненную модель tf.keras с помощью команды tf.keras.models.load_model ('yolov3.h5'), но сталкиваюсь с проблемой ниже

Traceback (most recent call last):
  File "detect.py", line 77, in <module>
    app.run(main)
  File "/home/akshay/Downloads/vir3/lib/python3.6/site-packages/absl/app.py", line 299, in run
    _run_main(main, args)
  File "/home/akshay/Downloads/vir3/lib/python3.6/site-packages/absl/app.py", line 250, in _run_main
    sys.exit(main(argv))
  File "detect.py", line 41, in main
    model =tf.keras.models.load_model('yolov3.h5')
  File "/home/akshay/Downloads/vir3/lib/python3.6/site-packages/tensorflow/python/keras/saving/hdf5_format.py", line 215, in load_model
    custom_objects=custom_objects)
  File "/home/akshay/Downloads/vir3/lib/python3.6/site-packages/tensorflow/python/keras/saving/model_config.py", line 55, in model_from_config
    return deserialize(config, custom_objects=custom_objects)
  File "/home/akshay/Downloads/vir3/lib/python3.6/site-packages/tensorflow/python/keras/layers/serialization.py", line 95, in deserialize
    printable_module_name='layer')
  File "/home/akshay/Downloads/vir3/lib/python3.6/site-packages/tensorflow/python/keras/utils/generic_utils.py", line 192, in deserialize_keras_object
    list(custom_objects.items())))
  File "/home/akshay/Downloads/vir3/lib/python3.6/site-packages/tensorflow/python/keras/engine/network.py", line 1241, in from_config
    process_node(layer, node_data)
  File "/home/akshay/Downloads/vir3/lib/python3.6/site-packages/tensorflow/python/keras/engine/network.py", line 1197, in process_node
    layer(flat_input_tensors[0], **kwargs)
  File "/home/akshay/Downloads/vir3/lib/python3.6/site-packages/tensorflow/python/keras/engine/base_layer.py", line 612, in __call__
    outputs = self.call(inputs, *args, **kwargs)
  File "/home/akshay/Downloads/vir3/lib/python3.6/site-packages/tensorflow/python/keras/layers/core.py", line 768, in call
    return self.function(inputs, **arguments)
  File "/home/akshay/Downloads/vir3/yolov3-tf2/yolov3_tf2/models.py", line 220, in <lambda>
    boxes_0 = Lambda(lambda x: yolo_boxes(x, anchors[masks[0]], classes),
TypeError: list indices must be integers or slices, not list