learning rate를 옵티마이저에서 셋팅하는 코드

model.compile(tf.keras.optimizers.RMSprop(learning_rate=0.001), loss='mse', metrics=['mse', 'mae'])
return model
model = build_model()
힐링아무의 코딩일기 힐코딩!!
'A.I > Deep Learning' 카테고리의 다른 글
딥러닝/ 에포크 시마다 가장 좋은 모델을 저장하는 ModelCheckpoint (0) | 2022.06.16 |
---|---|
딥러닝/ Fine Tuning(파인 튜닝) 을 하기 위한 코드와 설명 (0) | 2022.06.16 |
딥러닝/ Transer Learning을 하기 위한 코드와 설명 (0) | 2022.06.16 |
딥러닝/ EarlyStopping 라이브러리 사용법 (0) | 2022.06.13 |
Phthon(파이썬) 딥러닝 밸리데이션 데이터란 무엇이고, 코드에서 사용하는 방법 (0) | 2022.06.13 |
댓글