我正在寻找一个工作示例,其中我使用TF Hub映像模块,为自定义任务对其进行微调,并将其反向传播到Hub模块。然后保存此结果模型并重新加载以进行评估。
https://github.com/tensorflow/hub/blob/master/examples/image_retraining/retrain.py 但我认为它不能训练模块。
当从检查点恢复模型时,它会像往常一样工作。
模块变量设置为原始值的唯一时间是在其初始值设定项运行时。