问题描述
feature_extractor = hub.KerasLayer(MODULE_HANDLE,input_shape=IMAGE_SIZE + (3,),output_shape=[FV_SIZE])
Error: ---------------------------------------------------------------------------
TypeError Traceback (most recent call last)
<ipython-input-45-4be659037f32> in <module>()
1 feature_extractor = hub.KerasLayer(MODULE_HANDLE,2 input_shape=IMAGE_SIZE + (3,----> 3 output_shape=[FV_SIZE]
4 )
1帧 /usr/local/lib/python3.6/dist-packages/tensorflow_core/python/ops/variables.py in hash(self)
TypeError:如果启用了Tensor相等,则变量不可散列。相反,请使用tensor.experimental_ref()作为键。
解决方法
使用TensorFlow集线器(最新版本为0.10.0) 它将解决这个问题
mainAxisAlignment: MainAxisAlignment.spaceBetween