def test_mobilenet_v2_base_quantized_and_softmax_classifier(self):
    base_model = bases.MobileNetV2Base(quantize=True)
    head_model = heads.SoftmaxClassifierHead(BATCH_SIZE, BOTTLENECK_SHAPE, NUM_CLASSES)
    optimizer = optimizers.SGD(LEARNING_RATE)
    model = TransferModel(self.dataset_dir, base_model, head_model, optimizer)
    self.assertModelAchievesAccuracy(model, 0.8)
def test_mobilenet_v2_base_quantized_and_softmax_classifier(self):
    base_model = bases.MobileNetV2Base(quantize=True)
    head_model = heads.SoftmaxClassifierHead(BATCH_SIZE, BOTTLENECK_SHAPE, NUM_CLASSES)
    optimizer = optimizers.SGD(LEARNING_RATE)
    model = TransferModel(self.dataset_dir, base_model, head_model, optimizer)
    self.assertModelAchievesAccuracy(model, 0.8)