解决tensorflow 1 迁移到tensorflow 2中AttributeError类问题的思路

    tensorflow 2 相对于tensorflow 1 或keras存在较多的变化,弃用了很多方法,部分是改了接口,部分是完全弃用,这时会导致AttributeError错误等各类找不到方法的问题。解决此类问题有两种思路:

1、针对改了接口,但存在替代接口的可以在官方API文档中通过搜索查看,如果能查到的就是替代接口还存在,此时使用替代接口即可。例如原来的tf.keras.backend.get_session 或者keras.backend.get_session,可以通过这里提示的新接口tf.compat.v1.keras.backend.get_session替代即可。

 2、针对完全弃用的接口,只能找替代方法。比如低版本keras中多卡并行的multi_gpu_model在高版本中完全弃用了,可以使用替代方法tf.distribute.MirroredStrategy()。具体可以参考:

解决 AttributeError: module ‘tensorflow.keras.utils‘ has no attribute ‘multi_gpu_model‘_豆芽菜-CSDN博客

扩展

迁移tf1到tf2时,简单的方式就是:

import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

猜你喜欢

转载自blog.csdn.net/dou3516/article/details/120078744