天天看點

解決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替代即可。

解決tensorflow 1 遷移到tensorflow 2中AttributeError類問題的思路

 2、針對完全棄用的接口,隻能找替代方法。比如低版本keras中多卡并行的multi_gpu_model在高版本中完全棄用了,可以使用替代方法tf.distribute.MirroredStrategy()。具體可以參考:

解決 AttributeError: module ‘tensorflow.keras.utils‘ has no attribute ‘multi_gpu_model‘_豆芽菜-CSDN部落格

繼續閱讀