Python count_params примеры использования

Пример #1

Файл: mobilenet_edgetpu_v1_model_test.py Проект: shizukanaskytree/models

 def test_create_mobilenet_edgetpu(self):
     model = mobilenet_edgetpu_v1_model.MobilenetEdgeTPU()
     self.assertEqual(common_modules.count_params(model), 4092713)

Пример #2

 def test_create_mobilenet_edgetpu(self):
     model = mobilenet_edgetpu_v2_model.MobilenetEdgeTPUV2()
     self.assertEqual(common_modules.count_params(model), 6069657)