Keras Callback เป็นตัวอย่างสำหรับบันทึกโมเดลหลังจากทุกยุคหรือไม่

ใครช่วยกรุณาโพสต์ตัวอย่างที่ตรงไปตรงมาของ Keras โดยใช้การโทรกลับเพื่อบันทึกแบบจำลองหลังจากทุกยุค ฉันสามารถหาตัวอย่างของการบันทึกน้ำหนัก แต่ฉันต้องการที่จะสามารถบันทึกรูปแบบการทำงานอย่างสมบูรณ์หลังจากยุคการฝึกอบรมทุกครั้ง

การตั้งค่า 'save_weights_only' เป็น False ใน Keras callback 'ModelCheckpoint' จะบันทึกแบบเต็ม ตัวอย่างที่นำมาจากลิงก์ด้านบนนี้จะบันทึกแบบเต็มรูปแบบทุกช่วงเวลาโดยไม่คำนึงถึงประสิทธิภาพ:

keras.callbacks.ModelCheckpoint(filepath, monitor='val_loss', verbose=0, save_best_only=False, save_weights_only=False, mode='auto', period=1)

พบตัวอย่างเพิ่มเติมที่นี่รวมถึงการบันทึกเฉพาะรุ่นที่ปรับปรุงแล้วและโหลดรุ่นที่บันทึกไว้

ตรวจสอบให้แน่ใจว่าได้รวมตัวแปร epoch ใน filepath ของคุณ มิฉะนั้นโมเดลที่บันทึกของคุณจะถูกแทนที่หลังจากทุกยุค

filepath = "saved-model-{epoch:02d}-{val_acc:.2f}.hdf5"
checkpoint = ModelCheckpoint(filepath, monitor='val_acc', verbose=1, save_best_only=False, mode='max')

สำหรับตัวอย่างเพิ่มเติมโปรดดูที่นี่