diff --git a/onnx2keras/utils.py b/onnx2keras/utils.py index 9b967913..bc4dca2b 100644 --- a/onnx2keras/utils.py +++ b/onnx2keras/utils.py @@ -65,7 +65,6 @@ def check_torch_keras_error(model, k_model, input_np, epsilon=1e-5, change_order if isinstance(input_np, np.ndarray): input_np = [input_np.astype(np.float32)] - l__ = input_np[0].copy() input_var = [Variable(torch.FloatTensor(i)) for i in input_np] pytorch_output = model(*input_var) diff --git a/test/layers/convolutions/test_conv2d.py b/test/layers/convolutions/test_conv2d.py index f5a8f834..19b463fc 100644 --- a/test/layers/convolutions/test_conv2d.py +++ b/test/layers/convolutions/test_conv2d.py @@ -53,13 +53,3 @@ def test_conv2d_case1(change_ordering, kernel_size, padding, stride, bias, dilat @pytest.mark.parametrize('groups', [1, 2, 3]) def test_conv2d_case2(change_ordering, kernel_size, padding, stride, bias, dilation, groups): func(change_ordering, kernel_size, padding, stride, bias, dilation, groups) - -# @pytest.mark.parametrize('change_ordering', [False]) -# @pytest.mark.parametrize('kernel_size', [7]) -# @pytest.mark.parametrize('padding', [5]) -# @pytest.mark.parametrize('stride', [1]) -# @pytest.mark.parametrize('bias', [False]) -# @pytest.mark.parametrize('dilation', [2]) -# @pytest.mark.parametrize('groups', [2]) -# def test_conv2d_case2(change_ordering, kernel_size, padding, stride, bias, dilation, groups): -# func(change_ordering, kernel_size, padding, stride, bias, dilation, groups) diff --git a/test/utils.py b/test/utils.py index 285b1d3e..68c85ef2 100644 --- a/test/utils.py +++ b/test/utils.py @@ -1,10 +1,7 @@ import io -import numpy as np import torch import onnx -import collections -from typing import Union, Tuple from onnx2keras import onnx_to_keras, check_torch_keras_error