代碼報錯:ValueError: An operation has `None` for gradient. Please make sure that all of your ops have a

自定義Keras層後,代碼報錯,提示如標題,無法計算gradient。 原因:我是在別人模型的基礎上加上自己的代碼,修改了代碼之後,原來作者使用add_weight初始化的變量矩陣沒有刪除。導致錯誤。 如圖,沒有用到的變量需要及時刪除,否則在計算gradient時會出錯
相關文章
相關標籤/搜索