【发布时间】:2020-10-27 00:26:36
【问题描述】:
我一直在尝试训练我的皮肤类型分类模型,但它显示错误。
model_name = f"skintype_{batch_size}_{optimizer}"
tensorboard = tf.keras.callbacks.TensorBoard(log_dir=os.path.join("logs", model_name))
# saves model checkpoint whenever we reach better weights
modelcheckpoint = tf.keras.callbacks.ModelCheckpoint(model_name + "_{val_loss:.3f}.h5", save_best_only=True, verbose=1)
history = m.fit(train_ds, validation_data=valid_ds,
steps_per_epoch=n_training_samples // batch_size,
validation_steps=n_validation_samples // batch_size, verbose=1, epochs=30,
callbacks=[tensorboard, modelcheckpoint])
【问题讨论】:
标签: python artificial-intelligence classification