提交 5f0a8171 编写于 作者: W wanghaoshuang

Fix slim when compressing GoogleNet.

上级 7b0e8b41
......@@ -48,7 +48,7 @@ def compress(args):
# model definition
model = models.__dict__[args.model]()
if args.model is "GoogleNet":
if args.model == "GoogleNet":
out0, out1, out2 = model.net(input=image, class_dim=args.class_dim)
cost0 = fluid.layers.cross_entropy(input=out0, label=label)
cost1 = fluid.layers.cross_entropy(input=out1, label=label)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册