def testCreateModelInfo(self):
   did_raise_value_error = False
   try:
     retrain.create_model_info('no_such_model_name')
   except ValueError:
     did_raise_value_error = True
   self.assertTrue(did_raise_value_error)
   model_info = retrain.create_model_info('inception_v3')
   self.assertIsNotNone(model_info)
   self.assertEqual(299, model_info['input_width'])
Beispiel #2
0
 def testCreateModelInfo(self):
   did_raise_value_error = False
   try:
     retrain.create_model_info('no_such_model_name')
   except ValueError:
     did_raise_value_error = True
   self.assertTrue(did_raise_value_error)
   model_info = retrain.create_model_info('inception_v3')
   self.assertIsNotNone(model_info)
   self.assertEqual(299, model_info['input_width'])
 def testCreateModelInfoQuantized(self):
   # Test for mobilenet_quantized
   model_info = retrain.create_model_info('mobilenet_1.0_224')
   self.assertIsNotNone(model_info)
   self.assertEqual(224, model_info['input_width'])
Beispiel #4
0
 def testCreateModelInfoQuantized(self):
   # Test for mobilenet_quantized
   model_info = retrain.create_model_info('mobilenet_1.0_224')
   self.assertIsNotNone(model_info)
   self.assertEqual(224, model_info['input_width'])