Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
Address cr
Browse files Browse the repository at this point in the history
  • Loading branch information
reminisce committed Mar 23, 2018
1 parent f775050 commit a72955b
Show file tree
Hide file tree
Showing 3 changed files with 2 additions and 2 deletions.
1 change: 1 addition & 0 deletions example/quantization/launch_inference.sh
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@
# specific language governing permissions and limitations
# under the License.

set -ex

python imagenet_inference.py --symbol-file=./model/imagenet1k-resnet-152-symbol.json --param-file=./model/imagenet1k-resnet-152-0000.params --rgb-mean=0,0,0 --num-skipped-batches=50 --num-inference-batches=500 --dataset=./data/val_256_q90.rec

Expand Down
1 change: 1 addition & 0 deletions example/quantization/launch_quantize.sh
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@
# specific language governing permissions and limitations
# under the License.

set -ex

python imagenet_gen_qsym.py --model=imagenet1k-resnet-152 --calib-mode=none

Expand Down
2 changes: 0 additions & 2 deletions src/operator/quantization/quantized_conv.cc
Original file line number Diff line number Diff line change
Expand Up @@ -28,8 +28,6 @@
namespace mxnet {
namespace op {

// TODO(junwu): Reuse the InferShape function of convolution op after
// this pr is merged: https://github.com/apache/incubator-mxnet/pull/8302
bool QuantizedConvShape(const nnvm::NodeAttrs& attrs,
std::vector<TShape>* in_shape,
std::vector<TShape>* out_shape) {
Expand Down

0 comments on commit a72955b

Please sign in to comment.