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)