딥러닝 모델에 인덱스 사용방법

행동하는 개발자·2022년 11월 16일
0

PHM

목록 보기
13/43

모델을 전이학습시키려고 for 문으로 인덱스를 사용하여 불러오더니 타입이 모델에서 리스트로 변경되었다. 이 문제점을 다음과 같은 방법으로 해결하였다.

my_model = keras.models.load_model("./save_jh/200second_train_file/save_model_vib_bearing_model.hdf5")

base_model = keras.models.Sequential()

for i in my_model.layers[:-4]:
    base_model.add(i)

for layer in base_model.layers:
    layer.trainable = False

x = base_model.output    

인터넷에 있는 방법은 그냥 라이브러리 안의 모델을 불러오다보니 이런 디테일한 방법은 적어놓지 않아 애를 먹었다.

profile
끊임없이 뭔가를 남기는 사람

0개의 댓글