torchlens icon indicating copy to clipboard operation
torchlens copied to clipboard
Published 3 months ago • verified publisher icon johnmarktaylor91

FLOPs calculation function for forward propagation

Open whisperLiang opened this issue 5 months ago • 1 comments

Added FLOPs calculation function for forward propagation and back propagation per layer.

whisperLiang avatar Jul 08 '25 10:07 whisperLiang

Commit looks good—I will get back to this next week after a deadline but I expect should be smooth sailing to merge. Thanks so much for your great work on this, this has been on the wishlist for awhile :]

johnmarktaylor91 avatar Jul 09 '25 23:07 johnmarktaylor91