반응형
Deep Learning & Machine Learning/Keras2023. 10. 22. 05:45Keras 모델 전체 파라미터 개수 세기

Keras 모델의 전체 파라미터 개수를 출력하는 예제 코드입니다. 2023. 7. 19. 최초작성 from tensorflow.keras.applications import EfficientNetB0 # 모델을 로드합니다. model = EfficientNetB0(weights='imagenet') # 전체 파라미터 개수를 가져와 출력합니다. total_params = model.count_params() print(f'전체 파라미터 개수 = {total_params}') 실행해보면 EfficientNetB0 모델의 전체 파라미터 개수를 출력해줍니다. 전체 파라미터 개수 = 5330571

반응형
image