def test_create_mobilenet_edgetpu(self):
     model = mobilenet_edgetpu_v1_model.MobilenetEdgeTPU()
     self.assertEqual(common_modules.count_params(model), 4092713)
Пример #2
0
 def test_create_mobilenet_edgetpu(self):
     model = mobilenet_edgetpu_v2_model.MobilenetEdgeTPUV2()
     self.assertEqual(common_modules.count_params(model), 6069657)