Question

我想冻结模型并部署它们。首先，我使用InceptionV3遵循了1和2的slim附带的教程，这很好。然后，我尝试对从the provided url下载的resnet_v1_50执行相同的操作。

任何建议将不胜感激！

谢谢！

以下是步骤：

SQL Server

错误消息太长，我重定向了一个消息以引用它here。

Answer 1

如下行所示，导出推断图时，labels_offset应该设置为1。

InvalidArgumentError (see above for traceback): Assign requires shapes of both 
tensors to match. lhs shape= [1001] rhs shape= [1000]
     [[Node: save/Assign_265 = Assign[T=DT_FLOAT, _class= . 
["loc:@resnet_v1_50/logits/biases"], use_locking=true, validate_shape=true, 
_device="/job:localhost/replica:0/task:0/device:GPU:0"](resnet_v1_50/logits/biases, 
save/RestoreV2_265/_1)]]

命令类似于：

python export_inference_graph.py --model_name=resnet_v1_50 --labels_offset=1 --output_file=resnet_v1_50_inf_graph.pb

使用TF-slim在resnet_v1_50的冻结图上失败

1 个答案: