torchlens
torchlens copied to clipboard
FLOPs calculation function for forward propagation
Added FLOPs calculation function for forward propagation and back propagation per layer.
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 :]