tensorboard2使用指南。
from tensorflow.keras.callbacks import TensorBoard
tbCallBack = TensorBoard(log_dir="./log", histogram_freq=1, write_grads=True)
model.fit([x_train, x_train_masks], y_train,
batch_size=batch_size, epochs=epochs, validation_split=0.2, callbacks=[tbCa.
原创
2021-06-30 11:31:35 ·
171 阅读 ·
0 评论