Ошибка конвертера: ошибка TOCO. Смотрите консоль для информации - PullRequest
Новая
       10

Ошибка конвертера: ошибка TOCO. Смотрите консоль для информации

0 голосов
/ 19 марта 2019

Я новичок в тензорном потоке, я запускал свой первый пример в тензорном потоке, но обнаружил эту проблему. Как я могу это сделать

"Ошибка TOCO. Информацию см. В консоли. \ N% s \ n% s \ n"% (stdout, stderr)) 

ConverterError: TOCO failed. See console for info.
Traceback (most recent call last):`enter code here`
  File "D:\Software\Scripts\toco_from_protos-script.py", line 6, in <module>
    from tensorflow.lite.toco.python.toco_from_protos import main
  File "D:\Software\lib\site-packages\tensorflow\__init__.py", line 31, in <module>
    from tensorflow._api.v1 import audio
ImportError: cannot import name 'audio' from 'tensorflow._api.v1' (D:\Software\lib\site-packages\tensorflow\_api\v1\__init__.py)

Ниже мой код Python 

import tensorflow as tf    
import numpy as np    
from tensorflow import keras    
from tensorflow import lite    

model = keras.Sequential([keras.layers.Dense(units=1,input_shape=[1])])    
model.compile(optimizer='sgd',loss='mean_squared_error');    

xs=np.array([-1.0,0.0,1.0,2.0,3.0,4.0], dtype=float)    
ys= np.array([-3.0,-1.0,0.0,3.0,5.0,7.0],dtype=float)    

model.fit(xs,ys,epochs=20)    

print(model.predict([6.0]))    

#write out the keras save file    
keras_file="linear.h5"    
keras.models.save_model(model,keras_file)    

#convert to tensor flow     
converter = lite.TocoConverter.from_keras_model_file(keras_file)      
tflite_model = converter.convert()    

open('linear.tflite','wb').write(tflite_model)
...