Răsfoiți Sursa

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

YAMAMOTO Takashi 6 luni în urmă
părinte
comite
a29f3943ef

+ 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