Tensorflow hub fine-tune and evaluate

天大地大妈咪最大 提交于 2019-12-11 05:31:22

问题


I want to use tensorflow hub, to retrain one of its modules in my graph and then use that module, but my problem is when I set trainable = True and tags = {"train"} to create the module, I can not do an evaluation because of batch normalization layers. so as I read about this issue, I found that I should create also another graph for evaluation without setting tags = {"train"}. but I don't know how to restore variables from train graph into eval graph. I tried creating both modules with the same name and use reuse = True in the eval graph, but it wasn't helpful.

can you please help me with that? @arnoegw

来源:https://stackoverflow.com/questions/53849214/tensorflow-hub-fine-tune-and-evaluate

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!