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 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'])
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'])