def test_param_count(self): res = parameter_count(self.model) self.assertTrue(res[""], 41699936) self.assertTrue(res["backbone"], 26799296)
def test_param_count(self): res = parameter_count(self.model) self.assertTrue(res[""], 37915572) self.assertTrue(res["backbone"], 31452352)