Fix FLOPs calculation description in lecture_02.py#41

PrathamAditya wants to merge 1 commit into

stanford-cs336:mainstanford-cs336/lectures:mainfrom

PrathamAditya:mainPrathamAditya/lectures:mainCopy head branch name to clipboard