我正在寻找一个工作示例,我在其中使用 TF Hub 图像模块,针对自定义任务对其进行微调,并将其反向传播到集线器模块。然后保存这个结果模型并重新加载它以进行评估。
我看过https://github.com/tensorflow/hub/blob/master/examples/image_retraining/retrain.py但我不认为它训练模块。
从检查点恢复模型时,它照常工作。
如果您的模型正在从检查点恢复,那么模块变量也将从它恢复(而不是从原始模块加载)。
模块变量设置为原始值的唯一时间是运行它们的初始值设定项时。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句