错误NameError: name ‘tfa‘ is not defined

@NameError: name ‘tfa’ is not defined

出错代码

optimizer = tfa.optimizers.SWA(tf.keras.optimizers.SGD(lr=2.0), 100, 10)

解决问题

1 下载内置模块pip install tensorflow-addons

#pip install tensorflow-addons
import tensorflow_addons as tfa
optimizer = tfa.optimizers.SWA(tf.keras.optimizers.SGD(lr=2.0), 100, 10)

2 BUT 上述做法调用之后依旧出错，出错结果如下图。
因为安装的版本与tensforflow2.0版本不匹配。所以需要对应安装0.6.0版本的addons。