배우기 · 5

합성곱

이미지는 NCHW 다 — 배치·채널·높이·너비, torch 와 같은 배치. 자매 결속이 NHWC 를 쓴 것은 TF.js 의 제약을 피한 우회였고, 여기 커널에는 그 제약이 없어서 물려받지 않았다.

합성곱 하나

nn.Conv2d(inC, outC, kernel, stride, padding). 모양은 늘 쓰던 식으로 나오고, 확실히 하는 가장 빠른 방법은 찍어 보는 것이다.

작은 분류기

합성곱·활성·풀링을 두 번, 그다음 평탄화와 선형 머리. 처음 만드는 CNN 은 다 이 모양이고, 여기서는 수백 밀리초면 학습된다.

GPU 가 실제로 한 일

장치는 자기가 무엇을 얼마나 했는지 센다. 이 수들은 어댑터와 무관하다 — 코드 경로가 정하는 값이라, 벽시계 벤치가 못 도는 자리에서도 저장소의 cost 검사가 도는 이유가 그것이다.