自定义损失函数使用的隐藏坑 tensorflow2( 二 )


总结一下 , 当我们在模型中想用自定义的损失函数 , 不能直接传入fit函数 , 而是需要在train_step中手动传入 , 完成计算过程 。
到此这篇关于tensorflow2 自定义损失函数使用的隐藏坑的文章就介绍到这了,更多相关tensorflow2 自定义损失函数内容请搜索趣讯吧以前的文章或继续浏览下面的相关文章希望大家以后多多支持趣讯吧!

推荐阅读