|
|
|
@ -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):
|
|
|
|
|