본문 바로가기

Deep Learning

(labml.ai) AI 모델 구현 사이트 추천

https://nn.labml.ai/

 

labml.ai Annotated PyTorch Paper Implementations

 

nn.labml.ai

 

논문 검색하다가 우연히 알게된 사이트 입니다.

전 항상 논문을 보면 코드를 같이 봐야지 직성이 풀리는 스타일인데

깃허브에 올려놓은 구현들을 보면 환경 세팅같은 다른 부가적인 요소들도 같이 있어서

모델에 대한 내용만 이해하기가 굉장히 불편합니다.

하지만 이 사이트는 다른 부가적인 요소 없이 모델에 대한 코드만 구현해놔서 굉장히 편리합니다.

 

 

 

 

왼쪽에 모델 및 코드에 대한 설명을 적어놨고 오른쪽에 코드가 구현되어 있어

직관적으로 이해하기 쉽게 되어 있습니다.

 

모든 코드는 파이토치로 구현되어 있고

optimizer와 normaliztion 등 학습 기법들도 직접 코드로 확인하실 수 있습니다.

아쉬운점은 vgg net, inception net 같은 기본적인 모델에 대한 코드가 많이 없네요.