def test_param_with_reuse(self) -> None:
net = NetWithReuse(reuse=True)
count = parameter_count(net)
self.assertTrue(count[""], 90200)
self.assertTrue(count["conv2"], 100)
def test_param(self) -> None:
net = NetWithReuse()
count = parameter_count(net)
self.assertTrue(count[""], 180200)
self.assertTrue(count["conv2"], 90100)