Оболочка Python перезапускает ifoption для трассировки во время выполнения добавляется в model.compile при использовании keras - PullRequest
0 голосов
/ 15 октября 2018

Я использую python 3.5 tenorflow-gpu = 1.4.0.Сборка моего CNN с керасом.Модель имеет 2 обычных слоя и 2 плотных слоя.Следующий код работает без проблем.

model.compile(loss='mse', optimizer=optimizer)

Однако, если я хочу отследить память и время выполнения с помощью следующих строк:

run_options = tf.RunOptions(trace_level=tf.RunOptions.FULL_TRACE)
run_metadata = tf.RunMetadata()
model.compile(loss='mse', optimizer=optimizer,
      options=run_options,
      run_metadata=run_metadata)  

Оболочка Python перезапустится с первым вызовом

model.predict()

Я хочу знать, что вызывает это и как это предотвратить.

...