Tensorflow 2.1.0 - AttributeError: у модуля tenensflow нет атрибута RunMetadata - PullRequest
Новая
       28

Tensorflow 2.1.0 - AttributeError: у модуля tenensflow нет атрибута RunMetadata

1 голос
/ 10 февраля 2020

Я пытаюсь выяснить количество FLOPS, которые использует моя модель, используя этот код, который я получил в сети: 

def get_flops(model):
    run_meta = tf.RunMetadata()
    opts = tf.profiler.ProfileOptionBuilder.float_operation()

    # We use the Keras session graph in the call to the profiler.
    flops = tf.profiler.profile(graph=K.get_session().graph,
                                run_meta=run_meta, cmd='op', options=opts)

    return flops.total_float_ops  # Prints the "flops" of the model.


# .... Define your model here ....
print(get_flops(model))

Однако, выполнение этого кода дает мне эту ошибку: 

Traceback (most recent call last):
  File "/Users/Desktop/FYP/Code/Python/code/main.py", line 243, in <module>
    print(get_flops(model))
  File "/Users/Desktop/FYP/Code/Python/code/main.py", line 232, in get_flops
    run_meta = tf.RunMetadata()
AttributeError: module 'tensorflow' has no attribute 'RunMetadata'

Как я могу обойти эту ошибку? Я читал в Интернете, и единственная помощь, которую я получил, - это обновить мою версию tenorflow. Тем не менее, это самая обновленная версия.

...