search

keon / 3-min-pytorch

<펭귄브로의 3분 딥러닝, 파이토치맛> 예제 코드
https://book.naver.com/bookdb/book_detail.nhn?bid=15559613
MIT License
208 stars 246 forks source link

05-이미지_처리능력이_탁월한_CNN/resnet.ipynb #29

Open momstouch opened 3 years ago

momstouch commented 3 years ago

BasicBlock에서 첫번째 conv에만 인자로 받은 stride를 적용하고 두번째 conv는 stride = 1을 적용해야할 것 같아요. stride = 2를 두번 거쳐 w, h 가 /4이 되어 나오네용. 그래서 projection shortcut의 결과랑 dimension이 맞질 않습니다.