Avi Singhal

Results 1 issues of Avi Singhal

Hi, I went through the repo code and it seems only the FLOPs associated with nn.modules are going to be captured. But does the forward pass also affect the FLOPs?...