더북(TheBook)

7. 노트북 Commit

나만의 모델 학습 노트북이 완성됐다면 Save Version(Commit) 버튼을 눌러 노트북을 실행합니다. 그럼 내가 만든 코드가 즉시 캐글 플랫폼 내부에서 실행되고 모든 작업이 완료되면 실행 결과와 제출 파일이 표시됩니다. BERT-base 기준, 입력 데이터 차원(N, 220)일 때 1에포크 학습에 약 25분 남짓 소요됩니다. 이 과정에서 만들어진 모델 가중치 파일은 결과 제출을 위한 추론에 활용됩니다.

▲ 그림 7-17 TPU로 BERT를 학습한 로그

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.