>

에포크 수가 많은 keras 라이브러리를 사용하여 CNN 모델을 훈련합니다. 25 개의 에포크를 사용하여 처음으로 모델을 실행 한 다음 다음 코드 줄로 모델을 저장할 수 있습니까?

model.fit_generator(training_set,
                     steps_per_epoch = 100000,
                     epochs = 10,
                     validation_data = test_set,
                     validation_steps = 40000)
from keras.models import load_model
model.save('my_model.h5')

그런 다음 파이썬을 다시 시작하고 다음 코드와 같은 데이터 세트로 다음 15 개 에포크를 계속 실행합니다 :

model = load_model('my_model.h5')
model.fit_generator(training_set,
                     steps_per_epoch = 100000,
                     epochs = 15,
                     validation_data = test_set,
                     validation_steps = 40000)

교육을 계속하는 것으로 충분합니까? 또는 작업을 계속하려면 다른 단계를 수행해야합니다. 도움을 주셔서 감사합니다.

  • 답변 # 1

    예, 괜찮아요, model.save  가중치, 모델 아키텍처 및 최적화 상태를 저장하므로 문제없이 교육을 재개 할 수 있습니다.

관련 자료

  • 이전 PHP 다시로드 후 변수를 늘리는 방법
  • 다음 php - 발신자에게 회신 phpmailer