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

Fix slim when compressing GoogleNet.

上级 7b0e8b41
...@@ -48,7 +48,7 @@ def compress(args): ...@@ -48,7 +48,7 @@ def compress(args):
# model definition # model definition
model = models.__dict__[args.model]() 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) out0, out1, out2 = model.net(input=image, class_dim=args.class_dim)
cost0 = fluid.layers.cross_entropy(input=out0, label=label) cost0 = fluid.layers.cross_entropy(input=out0, label=label)
cost1 = fluid.layers.cross_entropy(input=out1, 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.
先完成此消息的编辑!
想要评论请 注册