创建张量流Lite模型

时间:2020-03-12 16:30:00

标签: tensorflow machine-learning tensorflow-lite

我正在使用模型个性化的tensorflow lite示例,它很好用。现在,我想创建更多的定制模型,但是意识到生成它的代码完全是手动的……这意味着它们创建了所有变量和反向执行步骤(keras代码可以选择,但它更大,更慢)。 Source code

我可以为更多的图层创建前向通过,但是不知道如何创建后向步骤(例如,通过relu激活)。 可以更熟练使用tensorflow的人是否会提供一些代码来简化代码,以便添加其他层以及反向传递?

0 个答案:

没有答案
相关问题