본문 바로가기

ML, DL/라이브러리

(4)
tensorflow, keras GPU 메모리 문제 https://twinparadox.tistory.com/606 TensorFlow, Keras GPU 메모리 문제(Out of Memory) 발생 시 시도해볼 방법 빅데이터 처리나 딥러닝을 하다보면 자연스럽게 마주하는 문제가 바로 메모리 문제다. 빅데이터 처리 같은 경우 그냥 파이썬에서 Numpy나 Dataframe을 사용하는 과정에서 발생하는 일이 흔하고, 그 twinparadox.tistory.com
Tensorflow checkpoint 출처 : https://eehoeskrap.tistory.com/343 [TensorFlow] .ckpt vs .pb vs .pbtxt 차이점 간단하게 말하자면 아래와 같다. ckpt 파일 모델의 변수(가중치)인 체크포인트 파일 pb 파일 모델의 변수 + 구조 (즉, 전체 그래프) 로 이루어진 바이너리 파일 pbtxt 파일 pb 파일을 읽을 수 있는 텍 eehoeskrap.tistory.com train1.ckpt.data-00000-of-00001 모델의 구조를 제외한 모든 변수들을 포함, 모델 복원시 메타파일과 데이터 파일을 이용 train1.ckpt.index 데이터파일과 메타파일을 올바르게 매핑하기 위한 내부적으로 필요한 인덱스 파일임 train1.ckpt.meta 메타 그래프를 포함, 변수 값을..
Pytorch
Tensorflow, Keras 출처 : https://wikidocs.net/32105 07) 케라스(Keras) 훑어보기 이 책에서는 딥 러닝을 쉽게 할 수 있는 파이썬 라이브러리인 케라스(Keras)를 사용합니다. 케라스는 유저가 손쉽게 딥 러닝을 구현할 수 있도록 도와주는 상위 레벨의 인터 ... wikidocs.net Keras 훑어보기 (Keras Sequential API) Sequential() 을 model 로 선언 한 뒤 model.add() 를 통계 layer 를 단계적으로 추가합니다. model = Sequential() Dense : Fully Conntect layer 를 추가합니다. model = Sequential() model.add(Dense(1, input_dim=3, activation='relu')..