Просмотр исходного кода

core/iwasm/libraries/wasi-nn/test: use the correct version of keras (#4383)

YAMAMOTO Takashi 8 месяцев назад
Родитель
Сommit
a29f3943ef
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      core/iwasm/libraries/wasi-nn/test/models/mult_outputs.py

+ 1 - 1
core/iwasm/libraries/wasi-nn/test/models/mult_outputs.py

@@ -3,7 +3,7 @@
 
 import tensorflow as tf
 import numpy as np
-from keras.layers import AveragePooling2D, Conv2D
+from tensorflow.keras.layers import AveragePooling2D, Conv2D
 
 from tensorflow.keras import Input, Model