안녕하세요! 혹시 [프로젝트 B] 관련해서 질문드려도 될까요???
7. Training 부분에서 model.fit을 구현하는 과정에 대한 질문입니다.
## 코드 시작 ##
model.fit(train_dataset, steps_per_epoch = train_len / batch_size,
validation_data = val_dataset,
validation_steps = val_len / batch_size,
epochs= max_epochs,
callbacks= [cp_callback])
## 코드 종료 ##
이렇게 시행했는데
Train for 100.0 steps, validate for 50.0 steps Epoch 1/20 1/100 [..............................] - ETA: 29s Epoch 00001: saving model to ./train/exp_cnn/cp-0001.ckpt 1/100 [..............................] - ETA: 41s
---------------------------------------------------------------------------
AttributeError Traceback (most recent call last)
<ipython-input-49-4bb7e2d08fe1> in <module>
4 validation_steps = val_len / batch_size,
5 epochs= max_epochs,
----> 6 callbacks= [cp_callback])
7 ## 코드 종료 ##라고 에러가 뜨네요.
cp_callback에 문제가 있나 싶어서 봐도
checkpoint_path = "./train/exp_cnn/cp-{epoch:04d}.ckpt"
checkpoint_dir = os.path.dirname(checkpoint_path)
## 코드 시작 ##
# 체크포인트 콜백 만들기
cp_callback = tf.keras.callbacks.ModelCheckpoint(checkpoint_path, save_weights_only = True, save_freq = 'epoch' , verbose=1) # tensorflow 2.0.0a 를 사용 중이라 save_freq 진행 X, 2.0.0로 업데이트 후 사용 가능했음
## 코드 종료 ##
checker.callback_check(cp_callback)
callback을 잘 정의하셨습니다! 이어서 진행하셔도 좋습니다
문제가 없다고 나와서 괴롭네요...ㅠㅠ
comment