|
|
|
@@ -49,13 +49,6 @@ class Net(nn.Cell): |
|
|
|
return out |
|
|
|
|
|
|
|
|
|
|
|
def test_shape_error(): |
|
|
|
""" for fuzz test""" |
|
|
|
in_str = "3 22222222222222222222222222264 3 64 64 222 222 3" |
|
|
|
with pytest.raises(ValueError): |
|
|
|
Net(in_str) |
|
|
|
|
|
|
|
|
|
|
|
class LeNet5(nn.Cell): |
|
|
|
""" LeNet5 definition """ |
|
|
|
|
|
|
|
@@ -98,7 +91,7 @@ class LeNet5(nn.Cell): |
|
|
|
return x |
|
|
|
|
|
|
|
|
|
|
|
def test_shape_error_2(): |
|
|
|
def test_shape_error(): |
|
|
|
""" for fuzz test""" |
|
|
|
in_str = "3 6 5 6 -6 5 16 5 5 120 120 84 84 3 2" |
|
|
|
with pytest.raises(ValueError): |
|
|
|
|