github-starred/cs249r_book
2
0
Fork 0
mirror of https://github.com/harvard-edge/cs249r_book.git synced 2026-03-11 17:49:25 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fix convolution FLOPs calculation in profiling example

Browse Source 
112 × 112 × 7 × 7 × 3 × 64 × 2 = 236,027,904 (not 235,012,096)
This commit is contained in:
Vijay Janapa Reddi
2026-01-27 08:38:18 -05:00
parent 91a75750df
commit d2b56498b3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinytorch/src/14_profiling/14_profiling.py
View File

@@ -1031,7 +1031,7 @@ Convolution FLOP Breakdown:
│ Kernel (out=64, in=3, kH=7, kW=7) │
│ ↓ │
│ Output size: (224×224) → (112×112) with stride=2 │
│ FLOPs = 112 × 112 × 7 × 7 × 3 × 64 × 2 = 235,012,096 FLOPs │
│ FLOPs = 112 × 112 × 7 × 7 × 3 × 64 × 2 = 236,027,904 FLOPs │
└────────────────────────────────────────────────────────────────┘
```