【发布时间】:2021-12-12 21:15:22
【问题描述】:
我正在使用tflite_model_maker。
from tflite_model_maker import model_spec
from tflite_model_maker import image_classifier
from tflite_model_maker.config import ExportFormat
from tflite_model_maker.config import QuantizationConfig
from tflite_model_maker.image_classifier import DataLoader
我想训练模型并想绘制图表以查看模型的性能但是这个没有history,
model = image_classifier.create(train_data,
model_spec = model_spec.get('efficientnet_lite4'),
validation_data=validation_data,
batch_size = 32,
epochs=200,
train_whole_model = True,
dropout_rate=0.25,
learning_rate = 0.01,
momentum = 0.9,
shuffle=True,
)
我该怎么办?
【问题讨论】:
-
@AloneTogether 是的,您的回答对我帮助很大。非常感谢你
-
不客气。
标签: python tensorflow classification