Skip to content

Commit eb78d0e

Browse files
obaileyw-unccjeremy-syn
authored andcommitted
Image classification model development: updated pretrained models to larger newly-trained model, fixed typo in model_converter.py which sliced characters from the model name that would fail convert_to_tflite.sh, removed old MD5 models
1 parent 9adec29 commit eb78d0e

6 files changed

Lines changed: 1 addition & 3 deletions

File tree

benchmark/training/image_classification/model_converter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
tfmodel_path = 'trained_models/' + model_name
1616
tfmodel = tf.keras.models.load_model(tfmodel_path)
1717
cifar_10_dir = 'cifar-10-batches-py'
18-
model_name = model_name[:-3]
18+
# model_name = model_name[:-3]
1919

2020
def representative_dataset_generator():
2121
train_data, train_filenames, train_labels, test_data, test_filenames, test_labels, label_names = \
Binary file not shown.

benchmark/training/image_classification/trained_models/pretrainedResnet.md5

Lines changed: 0 additions & 1 deletion
This file was deleted.
Binary file not shown.

benchmark/training/image_classification/trained_models/pretrainedResnet_quant.md5

Lines changed: 0 additions & 1 deletion
This file was deleted.
Binary file not shown.

0 commit comments

Comments
 (0)