신경망 설계 중 비용을 미리 가늠하는 오픈소스 도구

Tensey는 신경망을 눈으로 배치하며 설계하는 도구다. 설계하는 동안 텐서 모양이 서로 맞는지 확인하고, 매개변수 수와 , VRAM 사용량을 미리 계산한다.

잔차 연결이 맞지 않거나 층의 입력과 출력 크기가 어긋나는 문제를 GPU를 돌리기 전에 잡아낼 수 있다. 현재 63가지 연산을 지원하고, 모양 을 갖췄다.

완성한 설계는 실제로 실행되는 코드로 내보낼 수 있다. 코드는 GitHub에 공개되어 있고 로 사용할 수 있다.

핵심 포인트

  • 신경망 설계 중 텐서 모양 오류를 미리 확인한다.
  • 매개변수 수, , VRAM 사용량을 설계 단계에서 계산한다.
  • 잔차 연결 오류와 층 크기 불일치를 GPU 실행 전에 잡을 수 있다.
  • 63가지 연산을 지원하고 코드로 내보낼 수 있다.
  • GitHub에 공개된 오픈소스 도구다.
원문 보기