ソースを参照

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

YAMAMOTO Takashi 11 ヶ月 前
コミット
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