Skip to content

Commit

Permalink
removed unnecessary staff
Browse files Browse the repository at this point in the history
  • Loading branch information
Andrei Alkhouski committed Oct 29, 2020
1 parent beec10a commit 86a0e15
Show file tree
Hide file tree
Showing 3 changed files with 0 additions and 14 deletions.
1 change: 0 additions & 1 deletion onnx2keras/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down
10 changes: 0 additions & 10 deletions test/layers/convolutions/test_conv2d.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
3 changes: 0 additions & 3 deletions test/utils.py
Original file line number Diff line number Diff line change
@@ -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


Expand Down

0 comments on commit 86a0e15

Please sign in to comment.