2020年2月6日 星期四

[小技巧]在tensorflow2.0中沿用舊版的keras_contrib

最近在研究keras的cycleGan,將keras轉成tf.keras的版本時發生一個問題:
No module named 'keras_contrib'
看來似乎keras_contrib不存在於tensorflow2.0

取而代之的是tensorflow_addons
下pip指令:pip install tensorflow-addons

然後回到我上面的程式,將
from keras_contrib.layers.normalization.instancenormalization import InstanceNormalization
改成
from tensorflow_addons.layers import InstanceNormalization

就可以正常使用了

沒有留言:

張貼留言

有興趣或有疑問的歡迎提問與交流喔!!!