Cuiqing Li (李崔卿) tiandiao123
tiandiao123
/ optimize_bert.py
Created
July 1, 2021 22:01
— forked from icemelon/optimize_bert.py
Optimize the BERT model on CPUs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import time | |
| import argparse | |
| import numpy as np | |
| import mxnet as mx | |
| import gluonnlp as nlp | |
| import tvm | |
| from tvm import relay | |
| import tvm.contrib.graph_runtime as runtime | |
| def timer(thunk, repeat=1, number=10, dryrun=3, min_repeat_ms=1000): |