tensorboard在colab中的實現


colab現在自帶tensorboard的魔術方法了,用這個命令就能展示tensorboard

%reload_ext tensorboard
%tensorboard --logdir './callsbacks'



而在代碼中則需要寫明以下的部分
log_dir= './callsbacks'
if not os.path.exists(log_dir):
os.mkdir(log_dir)
output_model_file = os.path.join(log_dir,"best.h5")
callbacks = [
tf.keras.callbacks.TensorBoard(log_dir),
tf.keras.callbacks.ModelCheckpoint(output_model_file,save_beat_only = True),
tf.keras.callbacks.EarlyStopping(patience=5,min_delta=1e-3),
]
 
 
model.fit(trainX, trainY, batch_size=256, epochs=100, verbose=2, validation_split=0.1,callbacks = callbacks)
 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM