월간 데이콘 음성 중첩 데이터 분류 AI 경진대회

알고리즘 | Audio | 분류 | 제어 | KL Divergence

Prize : 100만원
584명 마감

Practice

Overview Data Code (share) Talk Leaderboard

Submission

Colab에서 TPU를 사용해서 학습하기 (Tesla P100 대비 약 4배 빠름)

It Shines Itself

2020.07.23 17:59 16,595 Views language

Colab의 TPU를 사용해서 학습한 코드입니다.

구글 클라우드 버킷 생성이 필요합니다.
https://cloud.google.com/storage/docs/creating-buckets?hl=ko

모델은 bbchip13님의 SampleCNN을 동일하게 사용하였습니다.
https://dacon.io/competitions/official/235616/codeshare/1316?page=1&dtype=recent&ptype=pub

Colab에서 사용시, Tesla P100 GPU의 경우 epoch당 약 120초가 소요되며,
TPU의 경우 epoch당 약 33초 가령 소요되었습니다.

제가 아직 머신러닝에 익숙하지 않아 부족한 부분이 있을 수 있습니다. 혹시 보시게 되면 지적 부탁드립니다.

Code