tensorflow2定義很多模型對內存的釋放

tensorflow2定義很多模型對內存的釋放 Keras作爲tensorflow的後端,管理全局狀態,它用來實現功能模型構建API和統一自動生成的層名。 如果需要在一個循環中創建許多模型,keras的全局狀態消耗越來越多的內存。可以通過調用clear_session()釋放全局狀態,這有助於避免舊模型和層的混亂,特別是在內存有限的時候。 在tf官網給出在循環中定義網絡之前清除內存的一個例子: 在
相關文章
相關標籤/搜索