问题
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