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)