Skip to content

Commit

Permalink
remove unnecessary model test
Browse files Browse the repository at this point in the history
  • Loading branch information
crazydemo committed Mar 3, 2022
1 parent 838a2ea commit 4e50729
Showing 1 changed file with 0 additions and 18 deletions.
18 changes: 0 additions & 18 deletions tests/python/contrib/test_dnnl.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
from gluoncv.model_zoo import get_model
import pytest
import itertools
import numpy as np
Expand Down Expand Up @@ -936,23 +935,6 @@ def get_graph(
run_and_verify_func(get_graph(relay.nn.max_pool3d, strides=(1, 1, 1)), run_module=run_module)


def run_and_verify_model(
model, run_module, input_shape=(1, 3, 224, 224), target="llvm", dtype="float32"
):
i_data = np.random.uniform(-1, 1, input_shape).astype(dtype)
block = get_model(model, pretrained=True)
mod, params = relay.frontend.from_mxnet(block, shape={"data": input_shape}, dtype=dtype)
run_and_verify(mod, i_data, params, target=target, run_module=run_module)


@pytest.mark.skip(reason="takes a long time for this test ")
def test_model(run_module, dtype="float32"):
run_and_verify_model("ResNet50_v1b", run_module, dtype=dtype)
run_and_verify_model("VGG11_bn", run_module, dtype=dtype)
run_and_verify_model("InceptionV3", run_module, input_shape=(1, 3, 300, 300), dtype=dtype)
run_and_verify_model("MobileNet1.0", run_module, dtype=dtype)


if __name__ == "__main__":
import sys

Expand Down

0 comments on commit 4e50729

Please sign in to comment.