Czy ktoś może napisać prosty przykład Keras wykorzystujący wywołanie zwrotne, aby zapisać model po każdej epoce? Potrafię znaleźć przykłady oszczędzania ciężarów, ale chcę móc zapisać w pełni funkcjonujący model po każdej epoce treningowej.
Czy ktoś może napisać prosty przykład Keras wykorzystujący wywołanie zwrotne, aby zapisać model po każdej epoce? Potrafię znaleźć przykłady oszczędzania ciężarów, ale chcę móc zapisać w pełni funkcjonujący model po każdej epoce treningowej.
Odpowiedzi:
Ustawienie „save_weights_only” na False w wywołaniu zwrotnym Keras „ModelCheckpoint” spowoduje zapisanie pełnego modelu; ten przykład wzięty z powyższego linku zapisze pełny model w każdej epoce, niezależnie od wydajności:
keras.callbacks.ModelCheckpoint(filepath, monitor='val_loss', verbose=0, save_best_only=False, save_weights_only=False, mode='auto', period=1)Więcej przykładów można znaleźć tutaj , w tym oszczędność tylko ulepszonych modeli oraz wczytywania zapisanych modeli.
Pamiętaj, aby uwzględnić zmienną epoki w ścieżce pliku. W przeciwnym razie zapisany model zostanie wymieniony po każdej epoce.
filepath = "saved-model-{epoch:02d}-{val_acc:.2f}.hdf5"
checkpoint = ModelCheckpoint(filepath, monitor='val_acc', verbose=1, save_best_only=False, mode='max')Aby uzyskać więcej przykładów, sprawdź tutaj .