diff --git a/caffe2/python/convnet_benchmarks.py b/caffe2/python/convnet_benchmarks.py index 89bbab6bf751a..bf9a1e3451e56 100644 --- a/caffe2/python/convnet_benchmarks.py +++ b/caffe2/python/convnet_benchmarks.py @@ -587,7 +587,8 @@ def create_resnet50_model_ops(model, loss_scale=1.0): "data", num_input_channels=3, num_labels=num_labels, - label="label", ) + label="label", + no_bias=True, ) prefix = model.net.Proto().name loss = model.net.Scale(loss, prefix + "_loss", scale=loss_scale) brew.accuracy(model, [softmax, "label"], prefix + "_accuracy")